Commit d4dacf45 authored by Decoupes Remy's avatar Decoupes Remy
Browse files

fix bug on classifier ouput format

parent 36fcfdfa
......@@ -155,7 +155,7 @@ if __name__ == '__main__':
candidate_labels = ["covid-19", "grippe aviaire", "AMR", "tiques", "autres"]
classifier_results = []
for i, tweets in tqdm(df_tfidf.iterrows()):
classifier_results.append(classifier(tweets["text"], candidate_labels))
classifier_results.append(classifier(tweets["text"], candidate_labels)["scores"])
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