from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from sklearn.cross_validation import train_test_split
iris = datasets.load_iris()
x = iris.data
y = iris.target
train_x, test_x, train_y, test_y = train_test_split(x, y, test_size=0.3, random_state=3)
dt_model = DecisionTreeClassifier()
dt_model.fit(train_x, train_y)
predict_y = dt_model.predict(test_x)
score = dt_model.score(test_x,test_y)
print(predict_y)
print(test_y)
print(‘得分:’,score)
用sklearn决策树对鸢尾花数据分类
最新推荐文章于 2024-04-07 08:59:43 发布