changing train/test split strategy to enable classification scripts