机器学习实战Ch03-决策树算法-【回归】预测隐形眼镜类型

玩泥巴写代码

于 2020-10-18 20:34:51 发布

阅读量223

点赞数

分类专栏：机器学习实战文章标签：决策树 python 算法

本文链接：https://blog.csdn.net/weixin_50883256/article/details/109149662

版权

机器学习实战专栏收录该内容

6 篇文章 0 订阅

订阅专栏

决策树算法：【回归】预测隐形眼镜类型

从《【分类】鱼》中调用creatTree
从《Matplotlib注释绘制树形图》中调用creatPlot

from treesfish import createTree
from treesplot import createPlot

fr = open('lenses.txt')
lenses = [inst.strip().split('\t') for inst in fr.readlines()]
lensesLabels = ['age', 'prescript', 'astigmatic', 'tearRate']
lensesTree=createTree(lenses,lensesLabels)
print(lensesTree)
createPlot(lensesTree)