From 9a7dc3bc52e03cc609c351fbbc96d3d1f6042f24 Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <moncla.ludovic@gmail.com>
Date: Thu, 19 Jan 2023 08:13:47 +0100
Subject: [PATCH] Update Classification_BertFineTuning.ipynb

---
 notebooks/Classification_BertFineTuning.ipynb | 20 ++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/notebooks/Classification_BertFineTuning.ipynb b/notebooks/Classification_BertFineTuning.ipynb
index 90aad20..6860b3d 100644
--- a/notebooks/Classification_BertFineTuning.ipynb
+++ b/notebooks/Classification_BertFineTuning.ipynb
@@ -227,6 +227,16 @@
         "!wget https://geode.liris.cnrs.fr/EDdA-Classification/datasets/test_set.tsv"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "!wget https://geode.liris.cnrs.fr/EDdA-Classification/datasets/training_set_superdomains.tsv\n",
+        "!wget https://geode.liris.cnrs.fr/EDdA-Classification/datasets/test_set_superdomains.tsv"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -244,8 +254,11 @@
       },
       "outputs": [],
       "source": [
-        "train_path = '../data/training_set.tsv'\n",
-        "test_path =  '../data/test_set.tsv'"
+        "#train_path = '../data/training_set.tsv'\n",
+        "#test_path =  '../data/test_set.tsv'\n",
+        "\n",
+        "train_path = '../data/training_set_superdomains.tsv'\n",
+        "test_path =  '../data/test_set_superdomains.tsv'"
       ]
     },
     {
@@ -298,7 +311,8 @@
       "outputs": [],
       "source": [
         "columnText = 'contentWithoutClass'\n",
-        "columnClass = 'ensemble_domaine_enccre'\n",
+        "#columnClass = 'ensemble_domaine_enccre'\n",
+        "columnClass = 'super_domain'\n",
         "\n",
         "maxOfInstancePerClass = 10000\n",
         "\n",
-- 
GitLab