Skip to content

erichson/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