Modify invalid class label and regenerate train data