"src/git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "f660b5fe4148fc3f5985b10c1fea969b695cb8de"
Commit fe9061fa authored by Guolin Ke's avatar Guolin Ke
Browse files

fix prediction in tree learner.

parent 07f709b9
......@@ -43,6 +43,7 @@ public:
}
void AddPredictionToScore(double* out_score) const override {
if (last_trained_tree_->num_leaves() <= 1) { return; }
#pragma omp parallel for schedule(static)
for (int i = 0; i < data_partition_->num_leaves(); ++i) {
double output = static_cast<double>(last_trained_tree_->LeafOutput(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