Changed the shape of loss

akrahdan · akrahdan · commit e20f16f98b64 · 2022-02-22T12:37:48.000-06:00
diff --git a/semevaltasks/classification/semeval_model.py b/semevaltasks/classification/semeval_model.py
@@ -1341,7 +1341,7 @@ def predict(self, to_predict):
         nb_eval_steps = 0
         preds = np.empty((len(to_predict), self.num_labels))
 
-        loss = np.empty((len(to_predict), self.num_labels))
+        loss = np.empty((len(to_predict), 1))
        
         out_label_ids = np.empty((len(to_predict)))
 
