cookiegg / DeepKoopmanWithControl

Soure code for Deep Koopman with Control

DeepKoopmanWithControl

Deep Koopman Operator with Control for Nonlinear Systems

Paper: https://arxiv.org/abs/2202.08004

Prediction

Control

Requirement

pytorch, gym

Environment

For gym environment, you should replace the gym env file with files in folder ./gym_env/

All the environments:

"DampingPendulum"
"DoublePendulum"
"Franka"
"Pendulum-v1"
"MountainCarContinous-v0"
"CartPole-v1"

Example

To train the network, you can just run

cd train/
python Learn_koopman_with_KlinearEig.py

To evaluate the prediction performance, you can utilize the notebooks in folder prediction/

To evaluate the control performance, you can utilize the notebooks in folder control/