基于决策树算法的钢板探伤预测模型优化 |
| |
作者姓名: | 王复越 任毅 赵坦 崔福祥 |
| |
作者单位: | 1. 海洋装备用金属材料及其应用国家重点实验室;2. 鞍钢集团钢铁研究院;3. 鞍钢股份有限公司鲅鱼圈钢铁分公司 |
| |
摘 要: | 采用决策树分类算法建立钢板探伤预测模型,结合冶金学原理选取精炼与连铸关键工艺特征属性,以轧后钢板探伤结果为目标标签,通过调整决策树最大深度、叶子最小样本数以及判定阈值对模型调优,经测试集验证:优化后的决策树模型对连铸板坯对应轧后钢板的探伤结果预测具有较好的预测效果,AUC值为0.848,且模型泛化能力较强,训练集与测试集AUC差值低于0.04。
|
关 键 词: | 数据挖掘 决策树算法 机器学习 连铸 探伤 |
|
|