Text Classification
sentence-transformers
PyTorch
Safetensors
setfit
mpnet
text-embeddings-inference
Instructions to use vgarg/setfit_zero_shot_test with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use vgarg/setfit_zero_shot_test with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("vgarg/setfit_zero_shot_test") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - setfit
How to use vgarg/setfit_zero_shot_test with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("vgarg/setfit_zero_shot_test") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds) - Notebooks
- Google Colab
- Kaggle
Add SetFit model
Browse files- model_head.pkl +2 -2
- pytorch_model.bin +1 -1
model_head.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a3888c878a8e42e0db88be4cc98bbbbb58b49aabe3579e4b8887e6ab860bb415
|
| 3 |
+
size 43951
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 438012518
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b904b9c82753f43ae6df5de76cba775d9bd6909f561caa4739308ed16ccb35e
|
| 3 |
size 438012518
|