Check correct dump format for gblinear.

dmlc · Sep 19, 2024 · 41183d6 · 41183d6
1 parent 15c6172
commit 41183d6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gbm/gblinear_model.h b/src/gbm/gblinear_model.h
@@ -125,7 +125,7 @@ class GBLinearModel : public Model {
         }
       }
       fo << std::endl << "    ]" << std::endl << "  }";
-    } else {
+    } else if (format == "text") {
       fo << "bias:\n";
       for (int gid = 0; gid < ngroup; ++gid) {
         fo << this->Bias()[gid] << std::endl;
@@ -136,6 +136,8 @@ class GBLinearModel : public Model {
           fo << (*this)[i][gid] << std::endl;
         }
       }
+    } else {
+      LOG(FATAL) << "Dump format `" << format << "` is not supported by the gblinear model.";
     }
     std::vector<std::string> v;
     v.push_back(fo.str());