データ分割でsklearnのtrain_test_split関数を使うように変更