机器学习实战——3.4 示例：使用决策树预测隐形眼镜类型

目录

1.数据集

2.加载数据集

3.预测隐形眼镜类型

---

1.数据集

2.加载数据集

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

3.预测隐形眼镜类型

myTree_lenses = trees.createTree(lenses, lensesLabels)
treePlotter.createPlot(myTree_lenses)

查看树结构：

 从上图我们可以发现，医生最多需要问四个问题就能确定患者需要佩戴哪种类型的隐形眼镜。