Update whitebox_utils/classifier.py

whitebox_utils/classifier.py

@@ -112,7 +112,6 @@ class MyClassifier():
             batch = self.tokenize(sent,idx=self.count)
             inputs = {"input_ids": batch[0], "attention_mask": batch[1], "token_type_ids":batch[2]}
             outputs = self.model(**inputs)[0] # get all logits
-            print(outputs)
-            output_ = [torch.softmax(output,dim=1)[0].detach().cpu().numpy() for output in outputs]
+            output_ = torch.softmax(outputs,dim=1)[0].detach().cpu().numpy()
             ret.append(output_)
         return np.array(ret)