fix.

dmlc · Jan 11, 2023 · 0940fe7 · 0940fe7
1 parent 17348ce
commit 0940fe7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc
@@ -152,8 +152,8 @@ void PredictByAllTrees(gbm::GBTreeModel const &model, const size_t tree_begin,
     } else {
       for (std::size_t i = 0; i < block_size; ++i) {
         auto t_predts = out_predt.Slice(predict_offset + i, linalg::All());
-        PredValueByOneTree<true>(thread_temp[offset + i], *tree.GetMultiTargetTree(), cats,
-                                 t_predts);
+        PredValueByOneTree<false>(thread_temp[offset + i], *tree.GetMultiTargetTree(), cats,
+                                  t_predts);
       }
     }
   }