Prediction improvement of compressive strength and strain of directionally solidified TiAl alloy based on training data size adjustment

Compressive strength and compressive strain, which are important mechanical properties of directionally solidified TiAl alloy, were predicted using machine learning algorithms, specifically Multiple Linear Regression (MLR), and Random Forest Regression (RFR). The input variables for the machine learning model were designated as the composition of the directionally solidified TiAl alloy, experimental parameters (pulling velocity, input power), and compression test parameters (strain rate, compression temperature). Compressive strength and compressive strain were designated as the output variables. Although the typical ratio of training and test data is 8:2, this study used different ratios of 9:1, 7:3, and 6:4 for machine learning, and excellent R2 values were obtained for all ratios. The feature importance, which can identify the factor that has the most influence on the output variables, was obtained through the RFR algorithm. According to the feature importance, temperature was found to have the greatest influence on compressive strength, while the Erbium (Er) element had the most significant influence on compressive strain. Through the results of feature importance, it was possible to quantitatively investigate the relationship between the Er element, a microalloying element that affects the microstructure of TiAl alloy, and the compressive properties. Furthermore, the study was conducted on which data ratio between training and test data is most suitable for predicting the compressive strength and strain of TiAl alloy. •The compressive strength and strain of directionally solidified (DS) TiAl alloy were predicted using the random forest regression (RFR) and multiple linear regression (MLR) algorithms.•The predicted values were observed by changing the machine learning training and test size ratio to 9:1, 8:2, 7:3, and 6:4.•Through the feature importance of RFR, the input variable values that have the most influence on the compressive strength and strain results were obtained, and as a result, the correlation between mechanical properties and microstructure was observed.•It was obtained that the 8:2 ratio was most suitable for predicting the compressive strength and strain of DS TiAl alloy.