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 { ...@@ -327,10 +327,10 @@ namespace LightGBM {
bool use_missing, bool zero_as_missing, bool use_missing, bool zero_as_missing,
const std::vector<double>& forced_upper_bounds) { const std::vector<double>& forced_upper_bounds) {
int na_cnt = 0; int na_cnt = 0;
int tmp_num_sample_values = 0; int non_na_cnt = 0;
for (int i = 0; i < num_sample_values; ++i) { for (int i = 0; i < num_sample_values; ++i) {
if (!std::isnan(values[i])) { if (!std::isnan(values[i])) {
values[tmp_num_sample_values++] = values[i]; values[non_na_cnt++] = values[i];
} }
} }
if (!use_missing) { if (!use_missing) {
...@@ -338,14 +338,14 @@ namespace LightGBM { ...@@ -338,14 +338,14 @@ namespace LightGBM {
} else if (zero_as_missing) { } else if (zero_as_missing) {
missing_type_ = MissingType::Zero; missing_type_ = MissingType::Zero;
} else { } else {
if (tmp_num_sample_values == num_sample_values) { if (non_na_cnt == num_sample_values) {
missing_type_ = MissingType::None; missing_type_ = MissingType::None;
} else { } else {
missing_type_ = MissingType::NaN; 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; bin_type_ = bin_type;
default_bin_ = 0; 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