Решение

from sklearn.tree import DecisionTreeClassifier
Outrageous Octopus