Fig. 1: Graph neural network training and testing.

a Active learning scheme to train a partial charge prediction graph neural network. Dropout Monte Carlo60 is used to compute the standard deviation (SD). b Test of generalization capabilities of partial charge predictions and uncertainties of the trained model under distribution shift.