Fix labels and migrate training level inside neural network