Commit
·
17f9d46
1
Parent(s):
a6829b1
bug fix
Browse files
upos.py
CHANGED
|
@@ -16,7 +16,6 @@ class BellmanFordTokenClassificationPipeline(TokenClassificationPipeline):
|
|
| 16 |
import numpy
|
| 17 |
if "logits" not in model_outputs:
|
| 18 |
return self.postprocess(model_outputs[0],**kwargs)
|
| 19 |
-
print(model_outputs["logits"].size())
|
| 20 |
m=model_outputs["logits"][0].numpy()
|
| 21 |
e=numpy.exp(m-numpy.max(m,axis=-1,keepdims=True))
|
| 22 |
z=e/e.sum(axis=-1,keepdims=True)
|
|
|
|
| 16 |
import numpy
|
| 17 |
if "logits" not in model_outputs:
|
| 18 |
return self.postprocess(model_outputs[0],**kwargs)
|
|
|
|
| 19 |
m=model_outputs["logits"][0].numpy()
|
| 20 |
e=numpy.exp(m-numpy.max(m,axis=-1,keepdims=True))
|
| 21 |
z=e/e.sum(axis=-1,keepdims=True)
|