サイズが大きいので、float化を諦めint[0-255]のまま格納
配列のコピーが起きないよう最適化
コード整理
とりあえずtrainだけをpickle化
全体が回るように