Update evaluate_train_rankings.py

evaluate_train_rankings.py

@@ -12,6 +12,7 @@ def load_json_file(file_path):
         return json.load(f)
 
 def main():
+    base_directory = os.getcwd()
     parser = argparse.ArgumentParser(description='Evaluate document ranking performance on training data')
     parser.add_argument('--pre_ranking', type=str, default='shuffled_pre_ranking.json',
                         help='Path to pre-ranking JSON file')
@@ -24,7 +25,7 @@ def main():
     parser.add_argument('--k_values', type=str, default='3,5,10,20',
                         help='Comma-separated list of k values for Recall@k')
     parser.add_argument('--base_dir', type=str, 
-                        default='datasets',
+                        default=f'{base_directory}/datasets',
                         help='Base directory for data files')
     args = parser.parse_args()