Enhancement of data for training neural network based constitutive models for geomaterials