Minor fixes

2022-07-29 11:05:37 +02:00 · 2022-07-29 11:05:37 +02:00 · 2b7188b1b8
commit 2b7188b1b8
parent 4d55499548
2 changed files with 3 additions and 4 deletions
--- a/docs/examples/scibert/train.ipynb
+++ b/docs/examples/scibert/train.ipynb
@ -246,7 +246,7 @@
    "    Dataset.from_dict({\"text\": X, \"label\": y})\n",
    "    .map(lambda v: tokenizer(v[\"text\"], truncation=True), batched=True)\n",
    "    .remove_columns(\"text\")\n",
-    "    .train_test_split(test_size=0.2, shuffle=True)\n",
+    "    .train_test_split(test_size=0.2, shuffle=True)  # test is actually validation\n",
    ")\n",
    "\n",
    "f1_score = load_metric(\"f1\")\n",