From d3200f70e10ae6342206ff00da72f96c244a0825 Mon Sep 17 00:00:00 2001
From: Nick Todd <nick.todd@t-s.qinetiq.com>
Date: Fri, 30 Aug 2024 10:23:34 +0100
Subject: [PATCH] #2844: Added evaluation stage to Ray notebooks.

---
 .../Training-an-RLLIB-MARL-System.ipynb          | 16 ++++++++++++++++
 .../notebooks/Training-an-RLLib-Agent.ipynb      | 16 ++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/src/primaite/notebooks/Training-an-RLLIB-MARL-System.ipynb b/src/primaite/notebooks/Training-an-RLLIB-MARL-System.ipynb
index 28f08edd..49801a2c 100644
--- a/src/primaite/notebooks/Training-an-RLLIB-MARL-System.ipynb
+++ b/src/primaite/notebooks/Training-an-RLLIB-MARL-System.ipynb
@@ -82,6 +82,22 @@
     "algo = config.build()\n",
     "results = algo.train()"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Evaluate the results"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "eval = algo.evaluate()"
+   ]
   }
  ],
  "metadata": {
diff --git a/src/primaite/notebooks/Training-an-RLLib-Agent.ipynb b/src/primaite/notebooks/Training-an-RLLib-Agent.ipynb
index 9d870192..2c35048d 100644
--- a/src/primaite/notebooks/Training-an-RLLib-Agent.ipynb
+++ b/src/primaite/notebooks/Training-an-RLLib-Agent.ipynb
@@ -74,6 +74,22 @@
     "algo = config.build()\n",
     "results = algo.train()\n"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Evaluate the results"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "eval = algo.evaluate()"
+   ]
   }
  ],
  "metadata": {