masa-su / pixyzoo

Implementation codes of deep generative models with Pixyz
102 stars 19 forks source link

NewtonianVAEの追加 #24

Closed ItoMasaki closed 1 year ago

ItoMasaki commented 1 year ago

創発システム研究室,M1の伊藤です.

現在,私が研究で利用しているNewtonianVAEをpixyzをベースに作成しました. pixyzooの中に入れてもらえれば幸いです.

原著論文の再現ができているかの確認を行えるように以下のプログラムを実装しています.

  1. データセットの作成
  2. モデルの学習
  3. 潜在空間とシミュレーション環境の間における座標に対する相関の確認
  4. シミュレーション環境での制御対象の操作

README.mdに上記4つのプログラムの実行方法を記述済みです.

不足しているものがあれば準備しますので,ご指摘をお願いします.

また,作成したpull request のマージ先が,masterブランチになっています.こちらからfeature/newtonian_vaeといったブランチ名の指定ができませんでしたので,どこのブランチにマージするのが適切かご指摘いただければ,再度pull requestの作成などを行います.

よろしくお願いします.