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库
最新推荐文章于 2022-06-27 19:20:26 发布