predict()
Возвращает предсказания для новых данных.
Сигнатура
estimator.predict(X)После fit возвращает предсказанные значения: метки классов для классификаторов или числа для регрессоров. Вызывать нужно на тех же признаках, что использовались при обучении.
y_pred = model.predict(X_test)
print(y_pred[:5])