决策树主函数
2022-05-12 22:55:11 8 举报
决策树主函数流程
作者其他创作
大纲/内容
得到生成的决策树后，利用for循环依次读取测试集test_set中的数据并输入字典，再调用Dtree中的子函数predict对测试集进行预测并输出测试结果test_predict
对于输出的测试结果test_predict，分别调用函数precision，recall，F1_score，对预测结果的精确率，召回率，F1分数进行计算，并输出最后结果。
调用子函数creat_data，读取数据，创建训练集和测试集并输出
fit函数中调用train函数对数据集进行训练并输出生成树
创建Dtree类型dt，并调用Dtree类中的函数fit对给定训练集train_data进行训练
子函数creat_data中先读取文件数据，然后进行删去数据和数据离散化，再调用split_train_test对读取的数据进行划分，成功输出训练集和测试集
main函数
0 条评论
回复 删除
下一页
