Hugging Face's logo

Spaces:
gorkaartola
/
Zero_Shot_Classifier_Tester_for_TP_FP
Runtime error

App Files Community
gorkaartola commited on
Commit
231b9bc
·
1 Parent(s): 4e01b20

Upload run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -44,7 +44,7 @@ def tp_tf_test(metric_selector, test_dataset, model_selector, queries_selector,
44
 
45
  results_test = pd.DataFrame()
46
  for query_data in queries_dataset:
47
- query = query_data['SDGquery']
48
  tokenized_test_dataset = test_dataset.map(tokenize_function, batched = True, fn_kwargs = {'prompt' : prompt, 'query' : query})
49
  columns_to_remove = test_dataset.column_names
50
  for column_name in ['label_ids', 'nli_label']:
 
44
 
45
  results_test = pd.DataFrame()
46
  for query_data in queries_dataset:
47
+ query = query_data['query']
48
  tokenized_test_dataset = test_dataset.map(tokenize_function, batched = True, fn_kwargs = {'prompt' : prompt, 'query' : query})
49
  columns_to_remove = test_dataset.column_names
50
  for column_name in ['label_ids', 'nli_label']: