Fig. 6: Training and testing datasets. | npj Computational Materials