train data classification method changed