Вы можете предсказать значения из любого классификатора машинного обучения, просто вызвав
Функция предсказания. Как код ниже.
from sklearn import tree
dtr = tree.DecisionTreeClassifier()
dtr.fit(X_train,y_train)
y_head_dtr = dtr.predict(X_test)
Сначала вы подгоняете свою модель, а затем с помощью функции прогнозирования вы получите прогноз.