我想用 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)