回答編集履歴

  train_test_splitは、

  Split arrays or matrices into random train and test subsets

 

  固定して分割したデータでやりたいのであれば、train_test_splitを使う必要はありませんし、使わない方がわかりやすいです。