Commit 512e0c34 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix warning.

parent cccd0587
...@@ -419,7 +419,7 @@ bool GBDT::TrainOneIter(const score_t* gradient, const score_t* hessian, bool is ...@@ -419,7 +419,7 @@ bool GBDT::TrainOneIter(const score_t* gradient, const score_t* hessian, bool is
UpdateScoreOutOfBag(new_tree.get(), curr_class); UpdateScoreOutOfBag(new_tree.get(), curr_class);
} else { } else {
// only add default score one-time // only add default score one-time
if (!class_need_train_[curr_class] && models_.size() < num_class_) { if (!class_need_train_[curr_class] && models_.size() < static_cast<size_t>(num_class_)) {
auto output = class_default_output_[curr_class]; auto output = class_default_output_[curr_class];
new_tree->Split(0, 0, BinType::NumericalBin, 0, 0, 0, new_tree->Split(0, 0, BinType::NumericalBin, 0, 0, 0,
output, output, 0, num_data_, 1); output, output, 0, num_data_, 1);
......
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