Skip to content

Commit a72a381

Browse files
committed
Add explanatory comment on filtering out XGBOOST_{REGRESSION,CLASSIFICATION}
1 parent ab37ab3 commit a72a381

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dataikuapi/dss/ml.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1361,6 +1361,7 @@ def get_enabled_algorithm_names(self):
13611361
:rtype: list of string
13621362
"""
13631363
algos = self.__class__.algorithm_remap
1364+
# Hide either "XGBOOST_CLASSIFICATION" or "XGBOOST_REGRESSION" which point to the same key "xgboost"
13641365
if self.mltask_settings["predictionType"] == "REGRESSION":
13651366
excluded_name = {"XGBOOST_CLASSIFICATION"}
13661367
else:

0 commit comments

Comments
 (0)