Update README.md
Browse files
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)
|