train data class