koopmanAE icon indicating copy to clipboard operation
koopmanAE copied to clipboard

Consistent Koopman Autoencoders

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