where x is the coordinates in the configuration space, y 1 (x) and y 1 (x) are the energy prediction of NN models, and w(y 1 , y 2 ) is the weight function to avoid unexpected sampling in high energy region. In this work, the weight function for NN training would also be the weight for the model uncertainty: