Merge pull request #28 from Alanhsiu/fix-example-bug

Fix Reward Calculation in example/2022-12-10-textrl-elon-musk.ipynb
voidful · May 9, 2024 · dc8568e · dc8568e
2 parents 5c98824 + c730f46
commit dc8568e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/example/2022-12-10-textrl-elon-musk.ipynb b/example/2022-12-10-textrl-elon-musk.ipynb
@@ -181,7 +181,7 @@
     "      if finish or len(predicted_list) >= self.env_max_length:\n",
     "        predicted_text = tokenizer.convert_tokens_to_string(predicted_list[0])\n",
     "        # sentiment classifier\n",
-    "        reward = sentiment(input_item[0]+predicted_text)[0][0]['score'] * 10\n",
+    "        reward = sentiment(input_item['input']+predicted_text)[0][0]['score'] * 10\n",
     "      return reward"
    ]
   },
@@ -330,7 +330,7 @@
   },
   "language_info": {
    "name": "python",
-   "version": "3.10.1"
+   "version": "3.9.9"
   },
   "vscode": {
    "interpreter": {
@@ -340,4 +340,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}