stMind

about Tech, Computer vision and Machine learning

2022-03-12から1日間の記事一覧

データセットをtrain/val/testに分割するコードをnumpyで簡潔に記述する

tl;dr numpy.splitを使って、aryを3つのsubarrayに分割する。 import numpy as np train, val, test = np.split(ary, [int(len(ary) * .6), int(len(ary) * .8)]) 簡単な説明 データをtrain/testに分割する時、scikit-learnのtrain_test_splitを使うことが多…