传统机器学习（一）：决策树在sklearn中的应用

一般流程

from sklearn import tree
from sklearn.datasets import load_wine
from sklearn.model_selection import train_test_split

#用红酒数据
wine = load_wine()
#分割训练集测试集
Xtrain,Xtest,Ytrain,Ytest = train_tese_split(wine.data, wine.target, test_size=0.3)
#训练模型

clf = tree.DecisionTreeClassifier(criterion = 'entropy'
                                ,random_state = 30
                                , splitter='random')
clf = clf.fit(Xtrain, Ytrain)
score = clf.score(Xtest, Ytest)

#决策树的可视化
feature_name = ['酒精','苹果酸','灰','灰的碱性','镁','总酚','类黄酮','非黄烷类酚类','花青素','颜色强度','色调','od280/od315稀释葡萄酒','脯氨酸']
import graphviz
dot_data = tree.export_graphviz(clf
                               ,out_file = None
                               ,feature_names= feature_name
                               ,cla