基于决策树模型的预测方法、装置和计算机设备

摘要

本申请涉及人工智能领域，提供一种基于决策树模型的预测方法、装置、计算机设备和存储介质，方法包括：获取每个企业相关的影响特征数据；获取对应的每个企业的类别结果；基于影响特征数据与类别结果建立样本数据集；基于训练数据集训练决策树模型获得初始决策树模型；对初始决策树模型进行剪枝得到最优决策树模型；对最优决策树模型进行精度评估，在评估通过后，将最优决策树模型作为目标决策树模型；将待预测的影响特征数据输入至目标决策树模型内，以输出对应的目标预测类别结果。本申请能够有效地实现了对于企业是否产生环境卫生违法行为的准确预测。本申请还可以应用于区块链领域，上述目标预测类别结果等数据可以存储于区块链上。