claritylab
/

zero-shot-explicit-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

21f97eb

•

1 Parent(s): 3871fe1

Update README.md

Files changed (1) hide show

README.md +7 -7

README.md CHANGED Viewed

@@ -36,11 +36,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)
+[[1.0987393e-03 9.9890125e-01]
+ [9.9988937e-01 1.1059999e-04]
+ [9.9986207e-01 1.3791372e-04]
+ [1.6576477e-03 9.9834239e-01]
+ [9.9990320e-01 9.6742726e-05]
+ [9.9894422e-01 1.0557596e-03]
+ [9.9959773e-01 4.0229000e-04]]
 ```