Gosse Minnema commited on
Commit
0c28e38
1 Parent(s): 5b0f5f5

Fix path for loading individual documents

Browse files
sociofillmore/common/analyze_text.py CHANGED
@@ -597,6 +597,8 @@ def analyze_single_document(doc_id, event_id, lome_model, dataset, texts_df, dee
597
  pred_file_path = f"output/migration/lome/multilabel/{lome_model}/pavia/{event_id}/lome_{doc_id}.comm.json"
598
  elif dataset == "femicides/olv":
599
  pred_file_path = f"output/femicides/lome/lome_0shot/multilabel/olv/{event_id}/lome_{doc_id}.comm.json"
 
 
600
  else:
601
  pred_file_path = f"output/{data_domain}/lome/lome_0shot/multilabel/{data_corpus}/{event_id}/lome_{doc_id}.comm.json"
602
  print(f"Analyzing file {pred_file_path}")
 
597
  pred_file_path = f"output/migration/lome/multilabel/{lome_model}/pavia/{event_id}/lome_{doc_id}.comm.json"
598
  elif dataset == "femicides/olv":
599
  pred_file_path = f"output/femicides/lome/lome_0shot/multilabel/olv/{event_id}/lome_{doc_id}.comm.json"
600
+ elif dataset == "femicides/rai":
601
+ pred_file_path = f"output/{data_domain}/lome/lome_0shot/multilabel/rai_ALL/{event_id}/lome_{doc_id}.comm.json"
602
  else:
603
  pred_file_path = f"output/{data_domain}/lome/lome_0shot/multilabel/{data_corpus}/{event_id}/lome_{doc_id}.comm.json"
604
  print(f"Analyzing file {pred_file_path}")