Skip to content

arvindrenga96/koopmanAE

 
 

Repository files navigation

Consistent Koopman Autoencoders

Research code that demonstrates consistent Koopman autoencoders on the nonlinear pendulum with no friction.

Here is an example:

python driver.py --dataset pendulum --folder results_back_pendulum --bottleneck 6 --backward 1

You can also create a baseline model:

python driver.py --dataset pendulum --folder results_pendulum --bottleneck 6 --backward 0

Use the following code to plot the results:

python plot_pred_error.py

Reference

Forecasting Sequential Data Using Consistent Koopman Autoencoders

About

Consistent Koopman Autoencoders

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.5%
  • Python 0.5%