高级检索
    谢少捷, 王伟, 何福善. 基于梯度提升决策树的特征筛选与钢卷力学性能预测[J]. 机械工程材料, 2021, 45(10): 104-110. DOI: 10.11973/jxgccl202110014
    引用本文: 谢少捷, 王伟, 何福善. 基于梯度提升决策树的特征筛选与钢卷力学性能预测[J]. 机械工程材料, 2021, 45(10): 104-110. DOI: 10.11973/jxgccl202110014
    XIE Shaojie, WANG Wei, HE Fushan. Feature Selection and Prediction of Mechanical Properties of Steel Coils Based on Gradient Boosting Decision Tree[J]. Materials and Mechanical Engineering, 2021, 45(10): 104-110. DOI: 10.11973/jxgccl202110014
    Citation: XIE Shaojie, WANG Wei, HE Fushan. Feature Selection and Prediction of Mechanical Properties of Steel Coils Based on Gradient Boosting Decision Tree[J]. Materials and Mechanical Engineering, 2021, 45(10): 104-110. DOI: 10.11973/jxgccl202110014

    基于梯度提升决策树的特征筛选与钢卷力学性能预测

    Feature Selection and Prediction of Mechanical Properties of Steel Coils Based on Gradient Boosting Decision Tree

    • 摘要: 基于冶金机理选取热镀锌钢卷基本建模特征,采用梯度提升决策树算法对其他化学元素特征进行筛选,结合网格搜索与交叉验证方法对模型参数进行优化,并利用模型分析不同特征对钢卷屈服强度的影响。结果表明:热镀锌钢卷力学性能预测建模的基本特征包括工艺参数特征、规格特征以及基本化学元素特征,对钢卷屈服强度影响较大的其他化学元素特征为氮、铝含量;模型参数优化后,在测试集上测得屈服强度的均方根误差为10.671 MPa,平均绝对误差为8.244 MPa,平均绝对百分误差为2.641%,模型预测精度比模型参数优化前的明显提高;当碳、硅、锰含量变化或热轧入轧温度变化时,钢卷屈服强度的变化幅度较大。

       

      Abstract: The basic modeling features of hot-dip galvanized steel coil were selected on the basis of metallurgical mechanism, and then other chemical element features were selected by gradient boosting decision tree algorithm. The model parameters were optimized by combining grid search with cross validation methods, and the effects of different features on yield strength of steel coils were analyzed by the model. The results show that the basic modeling features of mechanical property prediction model of hot-dip galvanized steel coils included process parameter features, specification features and basic chemical element features, and other chemical element features which had great influence on the yield strength of steel coils were nitrogen and aluminum content. After optimizing the model parameters, the root mean squared error of the yield strength measured on the test set was 10.671 MPa, the mean absolute error was 8.244 MPa, and the mean absolute percentage error was 2.641%. The prediction accuracy of the model was significantly higher than that before optimizing the model parameters. When the content of carbon, silicon or manganese or the hot rolling in-rolling temperature changed, the yield strength of steel coils changed greatly.

       

    /

    返回文章
    返回