Text Classification
setfit
Safetensors
sentence-transformers
mpnet
generated_from_setfit_trainer
Eval Results (legacy)
text-embeddings-inference
Instructions to use promforge/sbert-questionclassifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- setfit
How to use promforge/sbert-questionclassifier with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("promforge/sbert-questionclassifier") - sentence-transformers
How to use promforge/sbert-questionclassifier with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("promforge/sbert-questionclassifier") 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] - Notebooks
- Google Colab
- Kaggle
Sharuka Thirimanne commited on
Add SetFit model
Browse files- README.md +0 -0
- config.json +1 -1
- config_setfit.json +4 -1
- model.safetensors +1 -1
- model_head.pkl +2 -2
README.md
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "checkpoints/
|
| 3 |
"architectures": [
|
| 4 |
"MPNetModel"
|
| 5 |
],
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "checkpoints/step_25341",
|
| 3 |
"architectures": [
|
| 4 |
"MPNetModel"
|
| 5 |
],
|
config_setfit.json
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
{
|
| 2 |
-
"labels":
|
|
|
|
|
|
|
|
|
|
| 3 |
"normalize_embeddings": false
|
| 4 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"labels": [
|
| 3 |
+
0,
|
| 4 |
+
1
|
| 5 |
+
],
|
| 6 |
"normalize_embeddings": false
|
| 7 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 437967672
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4472458f2bd39954ada83d8bdab621284a049e952e2b8f8a6a8e08aad89f5b69
|
| 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:ab1ee698a2785bc53d5baec1ab0eba85b444eb4f03d64615c5853b19caee6ee8
|
| 3 |
+
size 7007
|