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