SamLowe commited on
Commit
f5ede76
1 Parent(s): a01ed1d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -84,12 +84,12 @@ input_feed_dict = {
84
  "attention_mask": [t.attention_mask for t in tokens_obj]
85
  }
86
 
87
- def sigmoid(_outputs):
88
- return 1.0 / (1.0 + np.exp(-_outputs))
89
-
90
  logits = model.run(output_names=output_names, input_feed=input_feed_dict)[0]
91
  # produces a numpy array, one row per input item, one col per label
92
 
 
 
 
93
  # Post-processing. Gets the scores per label in range.
94
  # Auto done by Transformers' pipeline, but we must do it manually with ORT.
95
  model_outputs = sigmoid(logits)
 
84
  "attention_mask": [t.attention_mask for t in tokens_obj]
85
  }
86
 
 
 
 
87
  logits = model.run(output_names=output_names, input_feed=input_feed_dict)[0]
88
  # produces a numpy array, one row per input item, one col per label
89
 
90
+ def sigmoid(_outputs):
91
+ return 1.0 / (1.0 + np.exp(-_outputs))
92
+
93
  # Post-processing. Gets the scores per label in range.
94
  # Auto done by Transformers' pipeline, but we must do it manually with ORT.
95
  model_outputs = sigmoid(logits)