export_graphviz()函数可选参数
时间: 2024-01-15 14:58:11 浏览: 52
`export_graphviz()`函数是用于将决策树模型导出为Graphviz格式的可视化工具,其可选参数如下:
- `out_file`: 字符串类型,输出文件名,可以是文件名、文件路径或文件对象,默认为None,表示输出到控制台。
- `feature_names`: 字符串列表类型,特征名称列表,表示每个特征的名称,默认为None,表示使用数字索引。
- `class_names`: 字符串列表类型,类别名称列表,表示每个类别的名称,默认为None,表示使用数字索引。
- `label`: 字符串类型,节点的标签类型,默认为'all',可以取值为'all'、'root'、'none',分别表示所有节点、根节点、没有节点标签。
- `filled`: 布尔类型,节点的填充颜色,默认为False,表示不填充。
- `leaves_parallel`: 布尔类型,叶子节点是否平行,默认为False,表示不平行。
- `impurity`: 布尔类型,每个节点的不纯度,默认为True,表示显示。
- `rotate`: 布尔类型,是否旋转树,默认为False,表示不旋转。
- `rounded`: 布尔类型,节点边框是否圆角,默认为False,表示不圆角。
- `special_characters`: 布尔类型,是否转义特殊字符,默认为True,表示转义。
- `precision`: 整数类型,输出浮点数的精度,默认为3。
- `max_depth`: 整数类型,树的最大深度,默认为None,表示不限制深度。
- `node_ids`: 布尔类型,是否显示节点ID,默认为False,表示不显示。
- `proportion`: 布尔类型,是否显示占比,默认为False,表示不显示。
阅读全文