weixin_42444020的博客

原创 python编程实现决策树算法

最近布置了个课堂作业，用python实现决策树算法。整了几天勉勉强强画出了棵歪脖子树，记录一下。大体思路：1.创建决策树My_Decision_Tree类，类函数__init__()初始化参数、fit()进行决策树模型训练、predict()进行预测、evaluate()进行模型评估、save_model()保存模型(csv格式)、load_model()加载模型、show_tree()使用Pillow库绘制决策树以及其他一些封装的功能函数；2.最佳划分点的度量通常有Gini值、Entropy.