Unverified Commit c5391c97 authored by Zhuyi Xue's avatar Zhuyi Xue Committed by GitHub
Browse files

renamed tmp_num_sample_values to non_na_cnt (#5521)

parent 45dd49ec
......@@ -327,10 +327,10 @@ namespace LightGBM {
bool use_missing, bool zero_as_missing,
const std::vector<double>& forced_upper_bounds) {
int na_cnt = 0;
int tmp_num_sample_values = 0;
int non_na_cnt = 0;
for (int i = 0; i < num_sample_values; ++i) {
if (!std::isnan(values[i])) {
values[tmp_num_sample_values++] = values[i];
values[non_na_cnt++] = values[i];
}
}
if (!use_missing) {
......@@ -338,14 +338,14 @@ namespace LightGBM {
} else if (zero_as_missing) {
missing_type_ = MissingType::Zero;
} else {
if (tmp_num_sample_values == num_sample_values) {
if (non_na_cnt == num_sample_values) {
missing_type_ = MissingType::None;
} else {
missing_type_ = MissingType::NaN;
na_cnt = num_sample_values - tmp_num_sample_values;
na_cnt = num_sample_values - non_na_cnt;
}
}
num_sample_values = tmp_num_sample_values;
num_sample_values = non_na_cnt;
bin_type_ = bin_type;
default_bin_ = 0;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment