From ef6d2f8a2e81255eb8223b2fa7c578c1ea3f7ac1 Mon Sep 17 00:00:00 2001
From: stwerner97 <stefanwerner1997@gmail.com>
Date: Wed, 31 Jul 2024 22:26:30 +0200
Subject: [PATCH] adding citation file to Readme

---
 README.md | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/README.md b/README.md
index de13f31..3e94256 100644
--- a/README.md
+++ b/README.md
@@ -44,3 +44,19 @@ python pdecontrol/mbrl/script.py
     --rollout_length_schedule "{\"scheduler\": \"LinearScheduler\", \"steptype\": \"iteration\", \"start\": 0, \"stop\": 200, \"vmin\": 3, \"vmax\": 7}" --policy_train_steps_per_sample 10
     --surrogate_train_freq 500
 ````
+
+
+# Citation
+
+If you our code for ``model-based-pde-control`` in your work, please cite our paper:
+
+`````
+@inproceedings{werner2024numerical,
+  title={Numerical Evidence for Sample Efficiency of Model-Based Over Model-Free Reinforcement Learning Control of Partial Differential Equations},
+  author={Werner, Stefan and Peitz, Sebastian},
+  booktitle={2024 European Control Conference (ECC)},
+  pages={2965--2971},
+  year={2024},
+  organization={IEEE}
+}
+````
\ No newline at end of file