机器学习--决策树(ID3)算法案例

本文介绍了一个使用决策树(ID3)算法预测隐形眼镜类型的案例。首先,展示了如何测试决策树分类函数,然后说明如何将决策树存储到磁盘并从磁盘读取。最后,通过读取'lenses.txt'文件的数据,利用特征属性创建隐形眼镜决策树,以帮助确定不同患者的隐形眼镜类型。
摘要由CSDN通过智能技术生成

       在进行案例分析前,先对决策树算法的分类函数进行测试。考虑到构造决策树非常耗时,为了节省计算时间,最好能够在每次执行分类时调用已经构造好的决策树。这就需要利用python模块pickle序列化对象将决策树分类算法保存在磁盘中,并在需要的时候读取出来。

1、测试决策树分类算法性能

######################################
#功能:决策树的分类函数
#输入变量:input_tree, feat_labels, test_vec
# 决策树,分类标签,测试数据
#输出变量:class_label 类标签
######################################
def classify(input_tree, feat_labels, test_vec):
    first_str = input_tree.keys()[0]
    second_dict = input_tree[first_str]
    class_label = -1

    # index方法用于查找当前列表中第一个匹配first_str变量的索引
    feat_index = feat_labels.index(first_str)

    for key in second_dict.keys():
        if test_vec[feat_index] == key:
        

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值