Commit 2b1595ef authored by Decoupes Remy's avatar Decoupes Remy
Browse files

fix typo in gephi and change zeroshot labels

parent 33da2019
......@@ -148,7 +148,7 @@ if __name__ == '__main__':
# zeroshot classification
classifier = pipeline("zero-shot-classification",
model="digitalepidemiologylab/covid-twitter-bert-v2-mnli")
candidate_labels_fr = ["covid-19"]
candidate_labels_fr = ["grippe aviaire"]
# candidate_labels_fr = ["covid-19", "grippe aviaire", "AMR", "tiques", "autres"]
candidate_labels_en = ["avian influenza"]
# candidate_labels_en = ["covid-19", "avian influenza", "AMR", "tick borne", "others"]
......@@ -191,7 +191,7 @@ if __name__ == '__main__':
gephi["Label"] = gephi["tf_idf_terms"]
gephi["timeset"] = gephi["@timestamp"]
gephi = gephi[gephi["Target"].str.len() !=0] # filter out tweet that are not retweeted
gephi[["ID", "label", "Source", "Target", "Timestamp"]].to_csv(
gephi[["Id", "label", "Source", "Target", "Timestamp"]].to_csv(
"analysis-output/acquitaine_script_gephi.csv",
index=False
)
......
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