sklearn.tree export_graphviz 函数用法

export_graphviz 函数用于将决策树导出为 Graphviz 格式，可以使用 Graphviz 工具将其转换为可视化图形。函数的基本用法如下：

from sklearn.tree import export_graphviz
import graphviz

# 将决策树导出为 Graphviz 格式
dot_data = export_graphviz(
    decision_tree,  # 决策树模型
    out_file=None,  # 不输出到文件，直接返回 Graphviz 格式的字符串
    feature_names=feature_names,  # 特征名称列表
    class_names=class_names,  # 类别名称列表
    filled=True,  # 是否对节点进行着色
    rounded=True,  # 是否对节点进行圆角处理
    special_characters=True)  # 是否对特殊字符进行转义处理

# 使用 Graphviz 工具将 Graphviz 格式的字符串转换为可视化图形
graph = graphviz.Source(dot_data)
graph.view()

其中，`decision_tree` 是训练好的决策树模型，`feature_names` 是特征名称列表，`class_names` 是类别名称列表，`filled` 和 `rounded` 分别指定是否对节点进行着色和圆角处理，`special_characters` 指定是否对特殊字符进行转义处理。函数返回的是 Graphviz 格式的字符串，需要使用 Graphviz 工具将其转换为可视化图形。