様々なサイトのtrain_test_split を使ったコードをコピペして勉強していたらエラーが出た話

機械学習の勉強をしていますと、データを簡単に訓練データとテストデータに分けてくれる、train_test_split を使う機会がちょくちょくありまして。

いろんなサイトのコードをコピペして試していたら、次のようなエラーが出てしまいました。

ModuleNotFoundError: No module named 'sklearn.cross_validation'

どういうこと？と思って調べていたら、train_test_splitは、sklearnのバージョンによって入ってるモジュールが違うようですね。 皆さまもお気を付けください。

# sklearn 0.20 以前のモジュール
from sklearn.cross_validation import train_test_split 

# sklearn 0.20 以降のモジュール
from sklearn.model_selection import train_test_split 

参考にさせていただいたサイトはこちら www.haya-programming.com

ちなみに train_test_split を使った手法を ホールドアウト検証 というそうです。

newtechnologylifestyle.net

また、train_test_split の詳しい使い方は以下のサイトをご参照ください。 note.nkmk.me

以上になります、最後までお読みいただきありがとうございました。