kerasを使ったLSTMモデルの学習に用いるデータセットの作成

Question

テニスの打球方向の打球方向の予測をLSTMを用いて実装しようとしてるのですが学習に用いるデータセットをcsvファイルで作成したのですが、その際csvファイルには選手がボールをインパクトしたフレームから20フレーム前までの各フレームにおける選手のにおける選手の位置座標を各行に記しています。そのため20フレームをワンセットとしてそれを100打球分つまり2000行のcsvファイルを作成したのですが20行をワンセットとしてcsvファイルをデータセットとして読み込ませるにはどうすればよいでしょうか、色々調べてみましたがよくわかりませんでした。どなたか参考記事などありましたらご教授頂きたいです。

Answer 1

※pandasを用いた手法です。望みのデータセットにするにはdataframeから特定の形式に変換する必要があるかもしれません。
一度すべて読み込んでしまってから20行ずつ処理を走らせるという手段もありますが、chunksize指定で初めから分割して読み込ませるのはどうでしょうか。

reader = pd.read_csv(fname, chunksize=20)

このように書けば一回で２０行だけ読み込ませることができます。本来一度で読み込みきれない膨大なファイルを扱うための手法ですが今回やりたいことと合致すると思います。
こちらのサイトを参考にしてみてください。
https://sinhrks.hatenablog.com/entry/2014/11/21/231534

Comments

[map_app2]

ありがとうございます。試してみます。