GAIL and AIRL tutorials: report also std in results

HumanCompatibleAI · Sep 6, 2023 · ab6e0c3 · ab6e0c3
1 parent 4fc83be
commit ab6e0c3
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/docs/tutorials/3_train_gail.ipynb b/docs/tutorials/3_train_gail.ipynb
@@ -147,8 +147,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "print(\"Mean reward before training:\", np.mean(learner_rewards_before_training))\n",
-    "print(\"Mean reward after training:\", np.mean(learner_rewards_after_training))"
+    "print(\n",
+    "    \"Rewards before training:\",\n",
+    "    np.mean(learner_rewards_before_training),\n",
+    "    \"+/-\",\n",
+    "    np.std(learner_rewards_before_training),\n",
+    ")\n",
+    "print(\n",
+    "    \"Rewards after training:\",\n",
+    "    np.mean(learner_rewards_after_training),\n",
+    "    \"+/-\",\n",
+    "    np.std(learner_rewards_after_training),\n",
+    ")"
    ]
   }
  ],

diff --git a/docs/tutorials/4_train_airl.ipynb b/docs/tutorials/4_train_airl.ipynb
@@ -151,8 +151,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "print(\"Mean reward before training:\", np.mean(learner_rewards_before_training))\n",
-    "print(\"Mean reward after training:\", np.mean(learner_rewards_after_training))"
+    "print(\n",
+    "    \"Rewards before training:\",\n",
+    "    np.mean(learner_rewards_before_training),\n",
+    "    \"+/-\",\n",
+    "    np.std(learner_rewards_before_training),\n",
+    ")\n",
+    "print(\n",
+    "    \"Rewards after training:\",\n",
+    "    np.mean(learner_rewards_after_training),\n",
+    "    \"+/-\",\n",
+    "    np.std(learner_rewards_after_training),\n",
+    ")"
    ]
   }
  ],