-
Notifications
You must be signed in to change notification settings - Fork 21
Open
Description
Currently "objective" parameter for the XGBoostClassifier
is limited to "reg:squarederror", "reg:tweedie", "reg:linear", "reg:logistic", "binary:logistic", "binary:logitraw". And these values are even forced with validation code:
if kkey == "objective":
if not isinstance(parameters["objective"], str):
raise TypeError("'objective' must be of type str")
if parameters["objective"] not in [
"reg:squarederror",
"reg:tweedie",
"reg:linear",
"reg:logistic",
"binary:logistic",
"binary:logitraw",
]:
raise ValueError(
"""'objective' must either be 'reg:squarederror', """
"""'reg:tweedie', 'reg:linear', 'reg:logistic', """
"""'binary:logistic', or 'binary:logitraw'"""
)
This code was clearly added when XGBoost already supported multiclass classification. So why can't I use an objective like "multi:softmax"? Or maybe there is some workaround for the multiclass classification?
Metadata
Metadata
Assignees
Labels
No labels