Commit 79c7c09d authored by Guolin Ke's avatar Guolin Ke
Browse files

use assert to avoid unknow exception.

parent e0ec8661
......@@ -172,7 +172,9 @@ void GBDT::ResetTrainingData(const BoostingConfig* config, const Dataset* train_
// multi-class
std::vector<data_size_t> cnt_per_class(num_tree_per_iteration_, 0);
for (data_size_t i = 0; i < num_data_; ++i) {
++cnt_per_class[static_cast<int>(label[i])];
int index = static_cast<int>(label[i]);
CHECK(index < num_tree_per_iteration_);
++cnt_per_class[index];
}
for (int i = 0; i < num_tree_per_iteration_; ++i) {
if (cnt_per_class[i] == num_data_) {
......
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