Commit bc9d57b3 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix one bug in 4bits_bin .

parent 1bade1e2
...@@ -747,6 +747,9 @@ void DatasetLoader::ConstructBinMappersFromTextData(int rank, int num_machines, ...@@ -747,6 +747,9 @@ void DatasetLoader::ConstructBinMappersFromTextData(int rank, int num_machines,
// find local feature bins and copy to buffer // find local feature bins and copy to buffer
#pragma omp parallel for schedule(guided) #pragma omp parallel for schedule(guided)
for (int i = 0; i < len[rank]; ++i) { for (int i = 0; i < len[rank]; ++i) {
if (ignore_features_.count(start[rank] + i) > 0) {
continue;
}
BinType bin_type = BinType::NumericalBin; BinType bin_type = BinType::NumericalBin;
if (categorical_features_.count(start[rank] + i)) { if (categorical_features_.count(start[rank] + i)) {
bin_type = BinType::CategoricalBin; bin_type = BinType::CategoricalBin;
......
...@@ -214,6 +214,7 @@ public: ...@@ -214,6 +214,7 @@ public:
OrderedBin* CreateOrderedBin() const override { return nullptr; } OrderedBin* CreateOrderedBin() const override { return nullptr; }
void FinishLoad() override { void FinishLoad() override {
if (buf_.empty()) { return; }
int len = (num_data_ + 1) / 2; int len = (num_data_ + 1) / 2;
for (int i = 0; i < len; ++i) { for (int i = 0; i < len; ++i) {
data_[i] |= buf_[i]; data_[i] |= buf_[i];
......
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