Commit 36fcfdfa authored by Decoupes Remy's avatar Decoupes Remy
Browse files

add tqdm before zero shot classification

parent 6598c331
......@@ -154,7 +154,7 @@ if __name__ == '__main__':
model="BaptisteDoyen/camembert-base-xnli")
candidate_labels = ["covid-19", "grippe aviaire", "AMR", "tiques", "autres"]
classifier_results = []
for i, tweets in df_tfidf.iterrows():
for i, tweets in tqdm(df_tfidf.iterrows()):
classifier_results.append(classifier(tweets["text"], candidate_labels))
print(classifier_results)
df_tfidf[[candidate_labels]] = pd.DataFrame(classifier_results)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment