Как заставить XGBoost работать с reg:squaredlogerror? ⇐ Python

[Anonymous]

Я пытаюсь обучить модель XGBoost с целью «reg:squaredlogerror», но она всегда выводит константу. Простой пример:

Код: Выделить всё

import numpy as np
from xgboost import XGBRegressor

xgb1 = XGBRegressor(n_estimators = 60, objective='reg:squaredlogerror')
xgb1.fit(np.array([[1], [2], [3]]), np.array([1, 2, 3]))
xgb1.predict(np.array([[1], [2], [3]]))

выход:

Код: Выделить всё

array([1.8770156, 1.8770156, 1.8770156], dtype=float32)

То же самое происходит и с большими наборами данных. Как мне создать модель XGBoost, которая минимизирует ошибку квадратичного журнала и не является константой?