导出决策树
在决策树分类任务中,过拟合通常源于树过深。通过比较训练集与测试集的准确率,您可以发现结果存在过拟合。从树的可视化也能看出这一点。
在本练习中,您将把决策树导出为一个文本文件,之后可用于可视化。
本练习是课程的一部分
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.____")