手把手带你开启机器学习——决策树的理解与实践

点击上方“超哥的杂货铺”,轻松关注 决策树简介 决策树是一种有监督的机器学习算法,可以实现分类和回归任务,通常对数据有比较好的拟合效果。 决策树能够非常直观地提供分类规则,易于解释。 公众号后台回复“决策树”可获得本文全部代码和PDF版本,便于练习保存。 决策树训练和可视化 使用sklearn自带的iris数据集,训练一个决策树,并用graphviz将其可视化出来。代码如下面所示: 需要提前在电脑
相关文章
相关标签/搜索