Fix dataset variable in aggregate_score.py to use TREC instead of BOOKS

flaviosoriano · flaviosoriano · commit 2c7e18ccf723 · 2026-03-08T14:52:13.000-03:00
diff --git a/aggregate_score.py b/aggregate_score.py
@@ -3,7 +3,7 @@
 
 # 1. Listar todos os arquivos do diretório
 
-dataset = "BOOKS"
+dataset = "TREC"
 caminho_arquivos = f"resource/result/Aggregated_LLM_RetrieverBERT_{dataset}/Aggregated_LLM_RetrieverBERT_{dataset}_*.rts"
 arquivos = glob.glob(caminho_arquivos)
 
