决策树分类器:贷款预测模型及验证
该代码使用决策树分类器对一个客户是否会贷款进行判断。代码首先定义了特征('年龄', '有工作', '有自己的房子', '信贷情况')和训练数据集,然后将特征和标签进行编码。接着,使用决策树分类器模型进行训练,并将训练后的模型可视化输出为一棵决策树。最后,将可视化的决策树保存在本地。
验证:
根据代码提供的数据集和特征,可以手动计算出该决策树的分类过程:
-
根据'年龄'特征进行分类,如果为老年,则判断有无自己的房子,有则预测为'是',否则判断有无工作,有则预测为'否',否则预测为'否'。
-
如果年龄为中年,则判断有无自己的房子,有则判断信贷情况,如果为'非常好',则预测为'是',否则预测为'否';如果没有自己的房子,则预测为'否'。
-
如果年龄为青年,则判断有无工作,有则判断有无自己的房子,有则预测为'好',否则预测为'一般';如果没有工作,则预测为'否'。
根据以上分类过程,可以发现该决策树的分类结果与代码中训练的模型结果一致,因此代码流程正确。
原文地址: https://www.cveoy.top/t/topic/n6GN 著作权归作者所有。请勿转载和采集!