from sklearn.datasets import load_iris from sklearn.tree import DecisionTreeClassifier iris = load_iris() X = iris.data y = iris.target tree_clf = DecisionTreeClassifier() tree_clf.fit(X, y)