from sklearn.datasets import load_iris
from sklearn import tree
X, y = load_iris(return_X_y=True)
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, y)
tree.plot_tree(feature_names=feature_names
, class_names=[‘a’, ‘b’, ‘c’]
, filled=True
, rounded=True) # class_names不显示中文
sklearn中决策树的绘制 不用graphviz库

相关文章: