nicholasKluge
/

RewardModel

Text Classification

preference model

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Aug 29, 2023

Commit

2730b6c

•

1 Parent(s): 16d7419

Upload RewardModel.ipynb

Files changed (1) hide show

RewardModel.ipynb +1 -1

RewardModel.ipynb CHANGED Viewed

@@ -495,7 +495,7 @@
         "\n",
         "os.makedirs(\"RewardModel\", exist_ok=True)\n",
         "\n",
-        "dataset = load_dataset(\"nicholasKluge/test-reward-dataset\", split=\"english\")\n",
         "\n",
         "print(\"Dataset loaded.\")\n",
         "\n",

         "\n",
         "os.makedirs(\"RewardModel\", exist_ok=True)\n",
         "\n",
+        "dataset = load_dataset(\"nicholasKluge/reward-aira-dataset\", split=\"english\")\n",
         "\n",
         "print(\"Dataset loaded.\")\n",
         "\n",