sklearnのtest_splitについて

Question

sklearnの関数で訓練データとテストデータに分けるものがあります。
data_train, data_test, label_train,
label_test = train_test_split(digits.data, digits.target)
このように書けるのですがなぜ訓練データとテストデータに分けるだけなのに4つに分けているのでしょうか？
最近ディープラーニングについて勉強したばかりでこのことがよく分からず質問しました。
よろしくお願いします。

Answer 1

digits.data は特徴量データです。 digits.target は目的変数（教師データ＝模範解答）です。
特徴量データと目的変数の並び順は一致していて、特徴量1つめのラベル（＝目的変数＝教師データ）は1つめに入ってます。つまり digits.data[0] のラベルが digits.target［０］ ということになります。

特徴量を訓練データとテストデータに分けたら、同じようにラベルも訓練データとテストデータに分ける必要があります。
digits.data を２つに分けて、この分け方と同じ分け方で digits.target も２つに分けると言うことです。このため、結果は4つ得られます。

特徴量訓練データ, 特徴量テストデータ, ラベル訓練データ, ラベルテストデータ = train_test_split(特徴量データ, ラベルデータ)

こんなイメージです。

Comments

[-Sh1r0]

回答ありがとうございます。そのように分かれていたんですね！構造体などでラベルも一緒に入っていると思っていました