总结
本发明公开了一种基于XGBoost算法的质量预估方法,属于金属热处理技术领域。包括以下步骤:S101:进行测试以获取相应的输入参数和输出参数作为建模数据;S102:根据建模数据建立XGBoost模型的目标函数,预设模型参数,建立质量预估的XGBoost模型;S103:将建模数据划分为训练集和验证集,对建立的XGBoost模型进行训练和验证以获得质量预估模型;S104:将待预估的输入参数代入训练好的质量预估模型,获得输出参数的预测值。本发明提出的基于XGBoost算法的感应淬火质量预估方法针对感应淬火过程复杂的问题,采用数据驱动模型对淬火质量进行预估,模型准确率高,能为缩短工艺制定周期提供帮助。