Synced code [skip ci]

Author: ankane

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