Skip to content

Commit a7be955

Browse files
committed
fix: change logits to sigmoid
1 parent 7e08b85 commit a7be955

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

run_goemotions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -179,10 +179,10 @@ def evaluate(args, model, eval_dataset, mode, global_step=None):
179179
eval_loss += tmp_eval_loss.mean().item()
180180
nb_eval_steps += 1
181181
if preds is None:
182-
preds = logits.detach().cpu().numpy()
182+
preds = 1 / (1 + np.exp(-logits.detach().cpu().numpy())) # Sigmoid
183183
out_label_ids = inputs["labels"].detach().cpu().numpy()
184184
else:
185-
preds = np.append(preds, logits.detach().cpu().numpy(), axis=0)
185+
preds = np.append(preds, 1 / (1 + np.exp(-logits.detach().cpu().numpy())), axis=0) # Sigmoid
186186
out_label_ids = np.append(out_label_ids, inputs["labels"].detach().cpu().numpy(), axis=0)
187187

188188
eval_loss = eval_loss / nb_eval_steps

0 commit comments

Comments
 (0)