Fig. 6: Neural network model training and computational efficiency.

a Training loss and testing loss as the model trains. The code saves the model with the lowest loss on the test data set as the best model. Both of the y axis curves are shown on a logarithmic scale. b A comparison of computation time between simulation and deep learning on 20 different structures