Объект 'gradientboostedtreeslearner' не имеет атрибута 'predict' YDF ⇐ Python

[Anonymous]

Ниже код YDF версии 0.15.0

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

from ydf import GradientBoostedTreesLearner, Task
model = GradientBoostedTreesLearner(label="_TARGET", task=Task.REGRESSION)
_X = X.copy()
_X["_TARGET"] = y
model.train(_X)

model.predict(X)

Я получил объект 'gradientboostedtreeslearner', не имеющий атрибута 'predict'