clean up / js / test_train_split