基于机器学习的工模具钢硬度预测

Hardness is a major indicator of the quality of tool and mold steels (TMS). Analytic formulas of hardness versus compositions were proposed here by using hierarchical clustering (HC) and LASSO regressions, based on data of 79 brands of TMS. HC presents two large groups of TMS, which contain high and low concentrations of chromium. Then, LASSO regressions were applied on each group and the regression result with the lowest root mean square error in leave-one-out cross-validation was taken out as the theoretical prediction formula of hardness versus chemical composition feature. Furthermore, atomic features were selected from electronegativity, atomic radius, valence electron number, electron affinity, first ionization energy, etc. LASSO regressions of the data with the atomic features give another prediction formula. These results demonstrate the powerful ability of machine learning in the design of TMS.