├── .gitignore ├── README.md ├── deepc_hunt ├── __init__.py ├── controllers.py ├── dynamics.py ├── trainer.py └── utils.py ├── examples ├── data │ ├── recht_ud.csv │ ├── recht_yd.csv │ ├── rocket_ud.csv │ └── rocket_yd.csv ├── images │ ├── deepc │ │ ├── deepc_bad.png │ │ ├── deepc_bad_traj.png │ │ ├── deepc_good.png │ │ └── deepc_good_traj.png │ └── mpc │ │ ├── mpc_bad.png │ │ ├── mpc_bad_traj.png │ │ ├── mpc_good.png │ │ └── mpc_good_traj.png ├── linear_deepc.ipynb ├── reproduce_paper │ ├── check_progress.py │ ├── costs_dict.pkl │ ├── rocket.py │ ├── success_dict.pkl │ └── train_rocket.py ├── requirements.txt └── rocket.ipynb ├── setup.py └── videos ├── DeePC.png ├── DeePC ├── DeePC-episode-0.meta.json ├── DeePC-episode-0.mp4 └── DeePC_trained.mp4 ├── DeePC_bad-episode-0.mp4 ├── DeePC_trained.mp4 ├── MPC ├── MPC-episode-0.meta.json └── MPC-episode-0.mp4 ├── cartpole_demo.mp4 └── cartpole_init.mp4 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/README.md -------------------------------------------------------------------------------- /deepc_hunt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/deepc_hunt/__init__.py -------------------------------------------------------------------------------- /deepc_hunt/controllers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/deepc_hunt/controllers.py -------------------------------------------------------------------------------- /deepc_hunt/dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/deepc_hunt/dynamics.py -------------------------------------------------------------------------------- /deepc_hunt/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/deepc_hunt/trainer.py -------------------------------------------------------------------------------- /deepc_hunt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/deepc_hunt/utils.py -------------------------------------------------------------------------------- /examples/data/recht_ud.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/data/recht_ud.csv -------------------------------------------------------------------------------- /examples/data/recht_yd.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/data/recht_yd.csv -------------------------------------------------------------------------------- /examples/data/rocket_ud.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/data/rocket_ud.csv -------------------------------------------------------------------------------- /examples/data/rocket_yd.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/data/rocket_yd.csv -------------------------------------------------------------------------------- /examples/images/deepc/deepc_bad.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/deepc/deepc_bad.png -------------------------------------------------------------------------------- /examples/images/deepc/deepc_bad_traj.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/deepc/deepc_bad_traj.png -------------------------------------------------------------------------------- /examples/images/deepc/deepc_good.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/deepc/deepc_good.png -------------------------------------------------------------------------------- /examples/images/deepc/deepc_good_traj.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/deepc/deepc_good_traj.png -------------------------------------------------------------------------------- /examples/images/mpc/mpc_bad.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/mpc/mpc_bad.png -------------------------------------------------------------------------------- /examples/images/mpc/mpc_bad_traj.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/mpc/mpc_bad_traj.png -------------------------------------------------------------------------------- /examples/images/mpc/mpc_good.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/mpc/mpc_good.png -------------------------------------------------------------------------------- /examples/images/mpc/mpc_good_traj.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/images/mpc/mpc_good_traj.png -------------------------------------------------------------------------------- /examples/linear_deepc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/linear_deepc.ipynb -------------------------------------------------------------------------------- /examples/reproduce_paper/check_progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/reproduce_paper/check_progress.py -------------------------------------------------------------------------------- /examples/reproduce_paper/costs_dict.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/reproduce_paper/costs_dict.pkl -------------------------------------------------------------------------------- /examples/reproduce_paper/rocket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/reproduce_paper/rocket.py -------------------------------------------------------------------------------- /examples/reproduce_paper/success_dict.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/reproduce_paper/success_dict.pkl -------------------------------------------------------------------------------- /examples/reproduce_paper/train_rocket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/reproduce_paper/train_rocket.py -------------------------------------------------------------------------------- /examples/requirements.txt: -------------------------------------------------------------------------------- 1 | ipykernel 2 | seaborn 3 | tqdm 4 | matplotlib 5 | -------------------------------------------------------------------------------- /examples/rocket.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/examples/rocket.ipynb -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/setup.py -------------------------------------------------------------------------------- /videos/DeePC.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC.png -------------------------------------------------------------------------------- /videos/DeePC/DeePC-episode-0.meta.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC/DeePC-episode-0.meta.json -------------------------------------------------------------------------------- /videos/DeePC/DeePC-episode-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC/DeePC-episode-0.mp4 -------------------------------------------------------------------------------- /videos/DeePC/DeePC_trained.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC/DeePC_trained.mp4 -------------------------------------------------------------------------------- /videos/DeePC_bad-episode-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC_bad-episode-0.mp4 -------------------------------------------------------------------------------- /videos/DeePC_trained.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/DeePC_trained.mp4 -------------------------------------------------------------------------------- /videos/MPC/MPC-episode-0.meta.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/MPC/MPC-episode-0.meta.json -------------------------------------------------------------------------------- /videos/MPC/MPC-episode-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/MPC/MPC-episode-0.mp4 -------------------------------------------------------------------------------- /videos/cartpole_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/cartpole_demo.mp4 -------------------------------------------------------------------------------- /videos/cartpole_init.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michael-cummins/DeePC-Hunt/HEAD/videos/cartpole_init.mp4 --------------------------------------------------------------------------------