removed the other objective in the dict

StatMixedML · Dec 4, 2023 · 2fa3eac · 2fa3eac
1 parent 3b3cff4
commit 2fa3eac
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lightgbmlss/model.py b/lightgbmlss/model.py
@@ -83,10 +83,9 @@ def set_params(self, params: Dict[str, Any]) -> Dict[str, Any]:
         """
         params_adj = {"num_class": self.dist.n_dist_param,
                       "metric": "None",
-                      "objective": "None",
                       "random_seed": 123,
                       "verbose": -1,
-                      'objective': self.dist.objective_fn}
+                      "objective": self.dist.objective_fn}
         params.update(params_adj)
 
         return params