claritylab
/

zero-shot-implicit-binary-bert

Zero-Shot Classification

sentence-transformers

zeroshot_classifier

text-classification

Model card Files Files and versions Community

StefanH commited on May 17, 2023

Commit

44f1a90

·

1 Parent(s): 733b37c

Update README.md

Files changed (1) hide show

README.md +7 -7

README.md CHANGED Viewed

@@ -49,11 +49,11 @@ You can use the model like this:
 >>> logits = model.predict(query, apply_softmax=True)
 >>> print(logits)
-[[6.8812753e-04 9.9931192e-01]
- [9.9974447e-01 2.5556990e-04]
- [9.9978167e-01 2.1833177e-04]
- [1.6187031e-03 9.9838126e-01]
- [9.9965131e-01 3.4869535e-04]
- [9.9413908e-01 5.8608940e-03]
- [9.9685740e-01 3.1425431e-03]]
 ```

 >>> logits = model.predict(query, apply_softmax=True)
 >>> print(logits)
+[[7.3497969e-04 9.9926502e-01]
+ [9.9988127e-01 1.1870124e-04]
+ [9.9988961e-01 1.1033980e-04]
+ [1.9227572e-03 9.9807727e-01]
+ [9.9985313e-01 1.4685343e-04]
+ [9.9938977e-01 6.1021477e-04]
+ [9.9838030e-01 1.6197052e-03]]
 ```