├── .gitignore ├── LICENSE ├── README.md ├── analyze-2body.ipynb ├── analyze-3body.ipynb ├── analyze-pend.ipynb ├── analyze-pixels.ipynb ├── analyze-real.ipynb ├── analyze-spring.ipynb ├── experiment-2body ├── data.py └── train.py ├── experiment-3body ├── data.py └── train.py ├── experiment-pend ├── data.py ├── pendulum.png └── train.py ├── experiment-pixels ├── data.py └── train.py ├── experiment-real ├── data.py ├── pendulum.png └── train.py ├── experiment-spring ├── data.py ├── mass-spring.png └── train.py ├── figures ├── 3body-base-example.pdf ├── 3body-compare.pdf ├── 3body-energy-compare.pdf ├── 3body-hnn-example.pdf ├── autoencoder-hnn.pdf ├── energy-compare.pdf ├── latents-hnn.pdf ├── learned-phase-space.pdf ├── orbits-base-example.pdf ├── orbits-compare.pdf ├── orbits-dataset.pdf ├── orbits-hnn-energy-distribution.pdf ├── orbits-hnn-example.pdf ├── pixels-compare-labeled.gif ├── pixels-compare.gif.png ├── pixels-dataset.gif.png ├── pixels-truth.gif.png └── total-energy.pdf ├── hnn.py ├── nn_models.py ├── static ├── addenergy.gif ├── autoencoder-hnn.png ├── hnn-paper.pdf ├── integrate-latent-hnn.png ├── latents-hnn.png ├── learned-phase-space.png ├── orbits-compare.gif ├── overall-idea.png ├── pdfs │ ├── 3body-base-example.pdf │ ├── 3body-compare.pdf │ ├── 3body-energy-compare.pdf │ ├── 3body-hnn-example.pdf │ ├── 3body-results.pdf │ ├── 3body-total-energy.pdf │ ├── 3body-train-curves-long.pdf │ ├── autoencoder-hnn.pdf │ ├── blog-summary-pend.pdf │ ├── blog-summary-real.pdf │ ├── blog-summary-spring.pdf │ ├── energy-compare.pdf │ ├── integrate-latent-hnn.pdf │ ├── latents-hnn.pdf │ ├── learned-phase-space.pdf │ ├── lipson-integration.pdf │ ├── lipson.pdf │ ├── orbit-results.pdf │ ├── orbits-base-example.pdf │ ├── orbits-compare.pdf │ ├── orbits-dataset.pdf │ ├── orbits-hnn-energy-distribution.pdf │ ├── orbits-hnn-example.pdf │ ├── orbits-total-energy.pdf │ ├── overall-idea.pdf │ ├── pend-integration.pdf │ ├── pend.pdf │ ├── pendulum-compare-static.pdf │ ├── real.pdf │ ├── simple-tasks.pdf │ ├── spring-integration.pdf │ ├── spring.pdf │ └── total-energy.pdf ├── pend-compare.gif └── pendulum-dataset.gif.png └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/README.md -------------------------------------------------------------------------------- /analyze-2body.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-2body.ipynb -------------------------------------------------------------------------------- /analyze-3body.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-3body.ipynb -------------------------------------------------------------------------------- /analyze-pend.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-pend.ipynb -------------------------------------------------------------------------------- /analyze-pixels.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-pixels.ipynb -------------------------------------------------------------------------------- /analyze-real.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-real.ipynb -------------------------------------------------------------------------------- /analyze-spring.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/analyze-spring.ipynb -------------------------------------------------------------------------------- /experiment-2body/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-2body/data.py -------------------------------------------------------------------------------- /experiment-2body/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-2body/train.py -------------------------------------------------------------------------------- /experiment-3body/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-3body/data.py -------------------------------------------------------------------------------- /experiment-3body/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-3body/train.py -------------------------------------------------------------------------------- /experiment-pend/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-pend/data.py -------------------------------------------------------------------------------- /experiment-pend/pendulum.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-pend/pendulum.png -------------------------------------------------------------------------------- /experiment-pend/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-pend/train.py -------------------------------------------------------------------------------- /experiment-pixels/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-pixels/data.py -------------------------------------------------------------------------------- /experiment-pixels/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-pixels/train.py -------------------------------------------------------------------------------- /experiment-real/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-real/data.py -------------------------------------------------------------------------------- /experiment-real/pendulum.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-real/pendulum.png -------------------------------------------------------------------------------- /experiment-real/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-real/train.py -------------------------------------------------------------------------------- /experiment-spring/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-spring/data.py -------------------------------------------------------------------------------- /experiment-spring/mass-spring.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-spring/mass-spring.png -------------------------------------------------------------------------------- /experiment-spring/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/experiment-spring/train.py -------------------------------------------------------------------------------- /figures/3body-base-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/3body-base-example.pdf -------------------------------------------------------------------------------- /figures/3body-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/3body-compare.pdf -------------------------------------------------------------------------------- /figures/3body-energy-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/3body-energy-compare.pdf -------------------------------------------------------------------------------- /figures/3body-hnn-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/3body-hnn-example.pdf -------------------------------------------------------------------------------- /figures/autoencoder-hnn.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/autoencoder-hnn.pdf -------------------------------------------------------------------------------- /figures/energy-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/energy-compare.pdf -------------------------------------------------------------------------------- /figures/latents-hnn.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/latents-hnn.pdf -------------------------------------------------------------------------------- /figures/learned-phase-space.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/learned-phase-space.pdf -------------------------------------------------------------------------------- /figures/orbits-base-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/orbits-base-example.pdf -------------------------------------------------------------------------------- /figures/orbits-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/orbits-compare.pdf -------------------------------------------------------------------------------- /figures/orbits-dataset.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/orbits-dataset.pdf -------------------------------------------------------------------------------- /figures/orbits-hnn-energy-distribution.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/orbits-hnn-energy-distribution.pdf -------------------------------------------------------------------------------- /figures/orbits-hnn-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/orbits-hnn-example.pdf -------------------------------------------------------------------------------- /figures/pixels-compare-labeled.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/pixels-compare-labeled.gif -------------------------------------------------------------------------------- /figures/pixels-compare.gif.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/pixels-compare.gif.png -------------------------------------------------------------------------------- /figures/pixels-dataset.gif.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/pixels-dataset.gif.png -------------------------------------------------------------------------------- /figures/pixels-truth.gif.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/pixels-truth.gif.png -------------------------------------------------------------------------------- /figures/total-energy.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/figures/total-energy.pdf -------------------------------------------------------------------------------- /hnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/hnn.py -------------------------------------------------------------------------------- /nn_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/nn_models.py -------------------------------------------------------------------------------- /static/addenergy.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/addenergy.gif -------------------------------------------------------------------------------- /static/autoencoder-hnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/autoencoder-hnn.png -------------------------------------------------------------------------------- /static/hnn-paper.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/hnn-paper.pdf -------------------------------------------------------------------------------- /static/integrate-latent-hnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/integrate-latent-hnn.png -------------------------------------------------------------------------------- /static/latents-hnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/latents-hnn.png -------------------------------------------------------------------------------- /static/learned-phase-space.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/learned-phase-space.png -------------------------------------------------------------------------------- /static/orbits-compare.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/orbits-compare.gif -------------------------------------------------------------------------------- /static/overall-idea.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/overall-idea.png -------------------------------------------------------------------------------- /static/pdfs/3body-base-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-base-example.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-compare.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-energy-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-energy-compare.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-hnn-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-hnn-example.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-results.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-results.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-total-energy.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-total-energy.pdf -------------------------------------------------------------------------------- /static/pdfs/3body-train-curves-long.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/3body-train-curves-long.pdf -------------------------------------------------------------------------------- /static/pdfs/autoencoder-hnn.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/autoencoder-hnn.pdf -------------------------------------------------------------------------------- /static/pdfs/blog-summary-pend.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/blog-summary-pend.pdf -------------------------------------------------------------------------------- /static/pdfs/blog-summary-real.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/blog-summary-real.pdf -------------------------------------------------------------------------------- /static/pdfs/blog-summary-spring.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/blog-summary-spring.pdf -------------------------------------------------------------------------------- /static/pdfs/energy-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/energy-compare.pdf -------------------------------------------------------------------------------- /static/pdfs/integrate-latent-hnn.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/integrate-latent-hnn.pdf -------------------------------------------------------------------------------- /static/pdfs/latents-hnn.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/latents-hnn.pdf -------------------------------------------------------------------------------- /static/pdfs/learned-phase-space.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/learned-phase-space.pdf -------------------------------------------------------------------------------- /static/pdfs/lipson-integration.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/lipson-integration.pdf -------------------------------------------------------------------------------- /static/pdfs/lipson.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/lipson.pdf -------------------------------------------------------------------------------- /static/pdfs/orbit-results.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbit-results.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-base-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-base-example.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-compare.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-compare.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-dataset.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-dataset.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-hnn-energy-distribution.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-hnn-energy-distribution.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-hnn-example.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-hnn-example.pdf -------------------------------------------------------------------------------- /static/pdfs/orbits-total-energy.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/orbits-total-energy.pdf -------------------------------------------------------------------------------- /static/pdfs/overall-idea.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/overall-idea.pdf -------------------------------------------------------------------------------- /static/pdfs/pend-integration.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/pend-integration.pdf -------------------------------------------------------------------------------- /static/pdfs/pend.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/pend.pdf -------------------------------------------------------------------------------- /static/pdfs/pendulum-compare-static.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/pendulum-compare-static.pdf -------------------------------------------------------------------------------- /static/pdfs/real.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/real.pdf -------------------------------------------------------------------------------- /static/pdfs/simple-tasks.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/simple-tasks.pdf -------------------------------------------------------------------------------- /static/pdfs/spring-integration.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/spring-integration.pdf -------------------------------------------------------------------------------- /static/pdfs/spring.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/spring.pdf -------------------------------------------------------------------------------- /static/pdfs/total-energy.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pdfs/total-energy.pdf -------------------------------------------------------------------------------- /static/pend-compare.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pend-compare.gif -------------------------------------------------------------------------------- /static/pendulum-dataset.gif.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/static/pendulum-dataset.gif.png -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/greydanus/hamiltonian-nn/HEAD/utils.py --------------------------------------------------------------------------------