CENT2: Improved charge equilibration via neural network technique