hsaest commited on
Commit
f218389
1 Parent(s): ddb68c7

Update evaluataion/eval.py

Browse files
Files changed (1) hide show
  1. evaluataion/eval.py +2 -2
evaluataion/eval.py CHANGED
@@ -40,9 +40,9 @@ def statistics(commonsense_statistic):
40
  def eval_score(validation_or_test: str, file_path: str, TOKEN):
41
 
42
  if validation_or_test == 'validation':
43
- query_data_list = load_dataset('osunlp/TravelBenchEval','validation',token=TOKEN)['validation']
44
  elif validation_or_test == 'test':
45
- query_data_list = load_dataset('osunlp/TravelBenchEval','test',token=TOKEN)['test']
46
 
47
  query_data_list = [x for x in query_data_list]
48
  hardConstraint_statistic= {level:{day:[] for day in [3,5,7]} for level in ['easy','medium','hard']}
 
40
  def eval_score(validation_or_test: str, file_path: str, TOKEN):
41
 
42
  if validation_or_test == 'validation':
43
+ query_data_list = load_dataset('osunlp/TravelPlannerEval','validation',token=TOKEN)['validation']
44
  elif validation_or_test == 'test':
45
+ query_data_list = load_dataset('osunlp/TravelPlannerEval','test',token=TOKEN)['test']
46
 
47
  query_data_list = [x for x in query_data_list]
48
  hardConstraint_statistic= {level:{day:[] for day in [3,5,7]} for level in ['easy','medium','hard']}