Spaces:

gorkaartola
/

Zero_Shot_Classifier_Tester_for_TP_FP

Runtime error

gorkaartola commited on Aug 12, 2022

Commit

a3533fc

1 Parent(s): f63d4de

Upload run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -92,7 +92,7 @@ def tp_tf_test(metric_selector, test_dataset, model_selector, queries_selector,
 		raw_logits = raw.iloc[:,2:]
 		logits = np.zeros(shape=(len(raw_logits.index),len(classes)))
 		for i in range(len(classes)):
-			queries = queries_dataset.filter(lambda x: x['label_ids'] == i)['SDGquery']
 			logits[:,i]=raw_logits[queries].max(axis=1)
 		labels = raw[["dataset_labels","nli_labels"]]
 		labels = np.array(labels).astype(int)

 		raw_logits = raw.iloc[:,2:]
 		logits = np.zeros(shape=(len(raw_logits.index),len(classes)))
 		for i in range(len(classes)):
+			queries = queries_dataset.filter(lambda x: x['label_ids'] == i)['query']
 			logits[:,i]=raw_logits[queries].max(axis=1)
 		labels = raw[["dataset_labels","nli_labels"]]
 		labels = np.array(labels).astype(int)