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- config.json +1 -1
- model.safetensors +3 -0
- model_head.pkl +2 -2
config.json
CHANGED
|
@@ -19,6 +19,6 @@
|
|
| 19 |
"pad_token_id": 1,
|
| 20 |
"relative_attention_num_buckets": 32,
|
| 21 |
"torch_dtype": "float32",
|
| 22 |
-
"transformers_version": "4.
|
| 23 |
"vocab_size": 30527
|
| 24 |
}
|
|
|
|
| 19 |
"pad_token_id": 1,
|
| 20 |
"relative_attention_num_buckets": 32,
|
| 21 |
"torch_dtype": "float32",
|
| 22 |
+
"transformers_version": "4.35.0",
|
| 23 |
"vocab_size": 30527
|
| 24 |
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94f1c573e65cc1cb77d25ad2aa827da9616dbdb7040599d808cd7b753ac611c4
|
| 3 |
+
size 437967672
|
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:f1be59137a49714c084a5a701748b11c192a1c9605dc7d03a6bdbe6363df54e5
|
| 3 |
+
size 6991
|