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