开始使用免费开始使用

导出决策树

在决策树分类任务中,过拟合通常源于树过深。通过比较训练集与测试集的准确率,您可以发现结果存在过拟合。从树的可视化也能看出这一点。

在本练习中,您将把决策树导出为一个文本文件,之后可用于可视化。

本练习是课程的一部分

HR Analytics:用 Python 预测员工流失

查看课程

练习说明

  • sklearn.tree 子模块中导入 export_graphviz() 函数。
  • model 拟合到训练数据。
  • 将可视化结果导出到 tree.dot 文件。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import the graphical visualization export function
from sklearn.____ import export_graphviz

# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)

# Export the tree to a dot file
____(model,"tree.____")
编辑并运行代码