From fc35ceb93c0f40b977506a2012da8f962a37052d Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <moncla.ludovic@gmail.com>
Date: Mon, 16 Jan 2023 08:34:55 +0100
Subject: [PATCH] Update Classification_Zero-Shot-Learning.ipynb
---
notebooks/Classification_Zero-Shot-Learning.ipynb | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/notebooks/Classification_Zero-Shot-Learning.ipynb b/notebooks/Classification_Zero-Shot-Learning.ipynb
index f6050c8..4a7ab88 100644
--- a/notebooks/Classification_Zero-Shot-Learning.ipynb
+++ b/notebooks/Classification_Zero-Shot-Learning.ipynb
@@ -102,7 +102,7 @@
"outputs": [],
"source": [
"import pandas as pd\n",
- "\n",
+ "from tqdm import tqdm\n",
"from transformers import BartForSequenceClassification, BartTokenizer\n"
]
},
@@ -759,7 +759,7 @@
"pred_labels = []\n",
"prob_labels = []\n",
"\n",
- "for content in df[column_text].tolist():\n",
+ "for content in tqdm(df[column_text].tolist()):\n",
"\n",
" true_probs = zero_shot_prediction(content[:1024], classes)\n",
" \n",
--
GitLab