改变XGBoost目标

Question

我试图在python中使用XGBoost进行logistic回归。我这样称呼它

import numpy as np
from xgboost import XGBClassifier

x_train = np.array([[1], [2], [3], [4]])
y_train = np.array([0, .25, .75, 1])

params = {
    "objective": "reg:logistic"
}

model = XGBClassifier(**params)
model.fit(x_train, y_train)
print(model.objective)

这输出了一个“多重:软This”而不是“reg:逻辑”的目标。因此，它并不是在进行逻辑回归。如何确保XGBoost不切换目标？

Answer 1

import numpy as np
from xgboost import XGBRegressor

x_train = np.array([[1], [2], [3], [4]])
y_train = np.array([[0], [0.25], [0.75], [1]])


model = XGBRegressor()
model.fit(x_train, y_train)
print(model.objective)

Answer 2

解决上述问题的方法是使用XGBRegressor而不是XGBClassifier。只是把它换进去似乎奏效了。