Using the Atomic Simulation Environment (ASE) and DeePMD-kit , the dipole moments of each molecule are calculated. The polarization vector is directly computed from Lammps. Addressing the fluctuations in water's induced dipole moment due to electronic polarizability, the DW model captures dielectric relaxation more adequately by considering flexible dipole moments in different local environments, outperforming classical force fields .