fixing incorrect work with multi:softprob_oneapi

razdoburdin · Oct 12, 2022 · eb99b42 · eb99b42
1 parent 93661a1
commit eb99b42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/learner.cc b/src/learner.cc
@@ -642,7 +642,8 @@ class LearnerConfiguration : public Learner {
   void ConfigureObjective(LearnerTrainParam const& old, Args* p_args) {
     // Once binary IO is gone, NONE of these config is useful.
     if (cfg_.find("num_class") != cfg_.cend() && cfg_.at("num_class") != "0" &&
-        tparam_.objective != "multi:softprob") {
+        (tparam_.objective != "multi:softprob") &&
+        (tparam_.objective != "multi:softprob_oneapi")) {
       cfg_["num_output_group"] = cfg_["num_class"];
       if (atoi(cfg_["num_class"].c_str()) > 1 && cfg_.count("objective") == 0) {
         tparam_.objective = "multi:softmax";