Python scikit-learn决策树模型（一）简单实现

这篇文章简单总结一下，在scikit-learn库里面实现决策树的步骤。

步骤一：

利用pandas库导入数据，定义自变量和因变量：

注意：分类变量应该先转换为数值标签。


步骤二：

选择模型，训练模型：

这里是简单模拟，都是用的默认配置，没有对参数进行修改。


步骤三：

模型可视化

注意：做图形导出前，需要先安装Graphviz软件，并将软件的bin文件夹路径添加到系统路径。

然后我们就能看到我们的简单模型了：


也可以用来预测：


©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容