Spaces:

ronaldahmed
/

ccl_win

Sleeping

ronald commited on Apr 26, 2023

Commit

043db6a

1 Parent(s): 62ff976

coh mech

Files changed (1) hide show

ccl_win.py CHANGED Viewed

@@ -134,7 +134,7 @@ class ccl_win(evaluate.Measurement):
         model = AutoModelForSequenceClassification.from_pretrained(os.path.join(BASEDIR,dataset))
         model.to(device)
-        pred_list,len_by_sample = self.preprocess_adjacent_window(preds)
         scores = []
         n_preds = len(pred_list)

         model = AutoModelForSequenceClassification.from_pretrained(os.path.join(BASEDIR,dataset))
         model.to(device)
+        pred_list,len_by_sample = self.preprocess_adjacent_window(predictions)
         scores = []
         n_preds = len(pred_list)