train-test split for level and grade