Next, we discuss a model trained on solvent only fragments (Table ). In particular, we assess how well the model can reproduce solvent polarization effects that are neglected by statically parameterized classical force fields. We do this by calculating dipole moments of individual water molecules around ion solvation shells (Fig. ) and comparing results from the model to a classical BSC1 force field and to reference quantum calculations.