Fig. 2: Prediction from the ML model trained with different parameters. | npj Computational Materials