Synced code [skip ci]

ankane · ankane · commit 036a0338aac4 · 2025-10-18T12:50:53.000-07:00
diff --git a/lib/xgboost/model.rb b/lib/xgboost/model.rb
@@ -20,6 +20,8 @@ def save_model(fname)
 
     def load_model(fname)
       @booster = Booster.new(params: @params, model_file: fname)
+      config = JSON.parse(@booster.save_config)
+      load_model_attributes(config)
     end
 
     def feature_importances
@@ -29,5 +31,11 @@ def feature_importances
       total = scores.sum.to_f
       scores.map { |s| s / total }
     end
+
+    private
+
+    def load_model_attributes(config)
+      @objective = config["learner"]["objective"]["name"]
+    end
   end
 end
