使用 tree.export_graphviz 改变决策树的形状

问题描述 投票:0回答:0

我想用 tree.export_graphviz 将我的决策树框更改为圆形或菱形,但我不能。有解决办法吗?

实际上,基于 sklearn 站点没有针对此问题的解决方案。 sklearn.tree.export_graphviz 涉及:

sklearn.tree.export_graphviz(decision_tree, out_file=None, *, max_depth=None, feature_names=None, class_names=None, label='all', filled=False, leaves_parallel=False, impurity=True, node_ids=False, 比例=False, rotate=False, rounded=False, special_characters=False, precision=3, fontname='helvetica') 并且没有任何更改形状的代码。 我的代码是:

clf = tree.DecisionTreeClassifier(random_state=42)
dot_data = tree.export_graphviz(clf,  
                  filled=True, rounded=True,
                  special_characters=True,
                  feature_names = ['a','b','c','d','e'],
                  class_names=target_names 
                  ) 
graph = graphviz.Source(dot_data)

enter image description here

python scikit-learn graphviz decision-tree decisiontreeclassifier
© www.soinside.com 2019 - 2024. All rights reserved.