Push model using huggingface_hub.
Browse files- README.md +5 -6
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
@@ -97,12 +97,11 @@ metrics:
|
|
97 |
pipeline_tag: text-classification
|
98 |
library_name: setfit
|
99 |
inference: true
|
100 |
-
base_model: TaylorAI/bge-micro-v2
|
101 |
---
|
102 |
|
103 |
-
# SetFit
|
104 |
|
105 |
-
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification.
|
106 |
|
107 |
The model has been trained using an efficient few-shot learning technique that involves:
|
108 |
|
@@ -113,7 +112,7 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
113 |
|
114 |
### Model Description
|
115 |
- **Model Type:** SetFit
|
116 |
-
- **Sentence Transformer
|
117 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
118 |
- **Maximum Sequence Length:** 512 tokens
|
119 |
- **Number of Classes:** 2 classes
|
@@ -187,8 +186,8 @@ preds = model("I guess I was attracted to this film both because of the sound of
|
|
187 |
|
188 |
| Label | Training Sample Count |
|
189 |
|:---------|:----------------------|
|
190 |
-
| negative | 8 |
|
191 |
| positive | 8 |
|
|
|
192 |
|
193 |
### Training Hyperparameters
|
194 |
- batch_size: (16, 2)
|
@@ -211,7 +210,7 @@ preds = model("I guess I was attracted to this film both because of the sound of
|
|
211 |
### Training Results
|
212 |
| Epoch | Step | Training Loss | Validation Loss |
|
213 |
|:------:|:----:|:-------------:|:---------------:|
|
214 |
-
| 0.1111 | 1 | 0.
|
215 |
|
216 |
### Framework Versions
|
217 |
- Python: 3.11.13
|
|
|
97 |
pipeline_tag: text-classification
|
98 |
library_name: setfit
|
99 |
inference: true
|
|
|
100 |
---
|
101 |
|
102 |
+
# SetFit
|
103 |
|
104 |
+
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
|
105 |
|
106 |
The model has been trained using an efficient few-shot learning technique that involves:
|
107 |
|
|
|
112 |
|
113 |
### Model Description
|
114 |
- **Model Type:** SetFit
|
115 |
+
<!-- - **Sentence Transformer:** [Unknown](https://huggingface.co/unknown) -->
|
116 |
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
117 |
- **Maximum Sequence Length:** 512 tokens
|
118 |
- **Number of Classes:** 2 classes
|
|
|
186 |
|
187 |
| Label | Training Sample Count |
|
188 |
|:---------|:----------------------|
|
|
|
189 |
| positive | 8 |
|
190 |
+
| negative | 8 |
|
191 |
|
192 |
### Training Hyperparameters
|
193 |
- batch_size: (16, 2)
|
|
|
210 |
### Training Results
|
211 |
| Epoch | Step | Training Loss | Validation Loss |
|
212 |
|:------:|:----:|:-------------:|:---------------:|
|
213 |
+
| 0.1111 | 1 | 0.1572 | - |
|
214 |
|
215 |
### Framework Versions
|
216 |
- Python: 3.11.13
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 69565312
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0beabd96ada2a324b931a2817a3f40c3a0c21ead8baa03a639eb5becaf945e01
|
3 |
size 69565312
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3983
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dbbad3e711e55dc3e9890da52a4fd5dfd2cd25e0c1e033e759f2487d4bceeced
|
3 |
size 3983
|