gpt4 book ai didi

python - 如何通过tree.plot_tree设置 TreeView 中的列名?

转载 作者:行者123 更新时间:2023-11-30 09:31:23 25 4
gpt4 key购买 nike

我试图通过sklearn lib树画一棵树,但问题是列索引写在图中。

tree.plot_tree(clf_decision)

enter image description here

最佳答案

利用feature_names and class_names parameters :

from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()
clf = tree.DecisionTreeClassifier(random_state=0).fit(iris.data, iris.target)


tree.plot_tree(clf, feature_names=iris.feature_names, class_names=iris.target_names)

enter image description here

关于python - 如何通过tree.plot_tree设置 TreeView 中的列名?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56503325/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com