batchを取得する際に毎度ファイルから取得しているのため遅い

なので、あらかじめデータセットをメモリーに載せいておいてメモリーから
batchを取得するようにしてデータロードを高速化する