一种基于决策树的预测方法及装置

摘要

本发明公开了一种基于决策树的预测方法及装置，涉及数据处理领域，能够提高决策树模型和决策树模型测过程的可视化效果，该方法具体包括：根据特征属性对预存的产品缺陷数据集中的数据进行属性选择生成目标训练集；采用决策树算法对训练集进行训练生成决策树，其中，训练集属于目标训练集；对决策树进行压缩得到第一决策树；利用可视化技术显示第一决策树；从测试集中选取至少一条测试例依次输入第一决策树中进行测试，生成至少一条测试例的分类路径；利用可视化技术在第一决策树中显示至少一条测试例的分类路径。本发明应用于产品缺陷预测。

主权项

一种基于决策树的预测方法，其特征在于，包括：根据特征属性对预存的产品缺陷数据集中的数据进行属性选择生成目标训练集，其中，所述特征属性为所述产品缺陷数据集中产品的固有属性或缺陷属性；采用决策树算法对所述训练集进行训练生成决策树，其中，所述训练集属于所述目标训练集；对所述决策树进行压缩得到第一决策树；利用可视化技术显示所述第一决策树；从测试集中选取至少一条测试例依次输入所述第一决策树中进行测试，生成所述至少一条测试例的分类路径；利用所述可视化技术在所述第一决策树中显示所述至少一条测试例的分类路径。