pytorchでlstmを用いてビリヤードの玉の起動を、予測するコードを書いてみようと思ったのですが

pytorchでlstmを用いてビリヤードの玉の起動を、予測するコードを書いてみようと思ったのですが、わからないので助けていただけると嬉しいです。
現在、ビリヤードの玉の座標をデータフレームでdf=pd.DataFrame({'x':[1,2,3,...],'y':[1,2,3,...]})の用に用意してあります。
ここから、t=0〜10のデータを使ってt=13ぐらいのデータを予測しようとしているのですが、コードを上手く書くことができません。
そこでアドバイスや参考になるサイトを教えていただけたら嬉しいです。よろしくおねがいします。

No.1 回答者： osamuy 回答日時： 2021/10/06 20:09

やろうとしてるネタが良くないように思えます。

高校レベルの物理で考えて、玉の回転や摩擦係数などが軌道を決定するので、ビリヤード玉の座標データだけでは、まともにAI学習できないのでは。

Unityとかで普通に物理演算させるほうが、まだ正確なような。