fix choices

sb-ai-lab · Mar 25, 2024 · 5d4849c · 5d4849c
1 parent 3c1a7de
commit 5d4849c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparklightautoml/ml_algo/boost_lgbm.py b/sparklightautoml/ml_algo/boost_lgbm.py
@@ -397,7 +397,7 @@ def _get_default_search_spaces(self, suggested_params: Dict, estimated_n_trials:
  optimization_search_space["numLeaves"] = Uniform(low=4, high=255, q=1)
 
  if self.task.name == "binary" or self.task.name == "multiclass":
- optimization_search_space["isUnbalance"] = Choice(choices=[0, 1])
+ optimization_search_space["isUnbalance"] = Choice(options=[0, 1])
 
  if estimated_n_trials > 30:
  optimization_search_space["baggingFraction"] = Uniform(low=0.5, high=1.0)