从 sklearn.tree 导入 DecisionTreeClassifier

from sklearn.tree import DecisionTreeClassifier

定义训练集

X = [[0,7], [1,5], [2,4], [3,1], [4,3], [5,6], [6,7], [7,9], [8,11]] y = [0, 0, 0, 1, 1, 1, 2, 2, 2]

任务1：创建 DecisionTreeClassifier 对象

########## Begin ########## DT = DecisionTreeClassifier() ########## End ##########

任务2：调用 fit 函数执行训练过程

########## Begin ########## DT.fit(X, y) ########## End ##########

调用 predict 函数进行预测

pre_y = DT.predict([[1,1], [7,10], [8,12]])

打印结果

print(pre_y)