Commit
•
4b4ae69
1
Parent(s):
1c280e3
Add SetFit model
Browse files- README.md +3 -24
- config_setfit.json +15 -1
- model_head.pkl +2 -2
README.md
CHANGED
@@ -164,25 +164,11 @@ widget:
|
|
164 |
|
165 |
'
|
166 |
inference: false
|
167 |
-
model-index:
|
168 |
-
- name: SetFit with avsolatorio/GIST-small-Embedding-v0
|
169 |
-
results:
|
170 |
-
- task:
|
171 |
-
type: text-classification
|
172 |
-
name: Text Classification
|
173 |
-
dataset:
|
174 |
-
name: Unknown
|
175 |
-
type: unknown
|
176 |
-
split: test
|
177 |
-
metrics:
|
178 |
-
- type: accuracy
|
179 |
-
value: 0.0
|
180 |
-
name: Accuracy
|
181 |
---
|
182 |
|
183 |
# SetFit with avsolatorio/GIST-small-Embedding-v0
|
184 |
|
185 |
-
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0) as the Sentence Transformer embedding model. A
|
186 |
|
187 |
The model has been trained using an efficient few-shot learning technique that involves:
|
188 |
|
@@ -194,9 +180,9 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
194 |
### Model Description
|
195 |
- **Model Type:** SetFit
|
196 |
- **Sentence Transformer body:** [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0)
|
197 |
-
- **Classification head:** a
|
198 |
- **Maximum Sequence Length:** 512 tokens
|
199 |
-
|
200 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
201 |
<!-- - **Language:** Unknown -->
|
202 |
<!-- - **License:** Unknown -->
|
@@ -207,13 +193,6 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
207 |
- **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
|
208 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
209 |
|
210 |
-
## Evaluation
|
211 |
-
|
212 |
-
### Metrics
|
213 |
-
| Label | Accuracy |
|
214 |
-
|:--------|:---------|
|
215 |
-
| **all** | 0.0 |
|
216 |
-
|
217 |
## Uses
|
218 |
|
219 |
### Direct Use for Inference
|
|
|
164 |
|
165 |
'
|
166 |
inference: false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
167 |
---
|
168 |
|
169 |
# SetFit with avsolatorio/GIST-small-Embedding-v0
|
170 |
|
171 |
+
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0) as the Sentence Transformer embedding model. A OneVsRestClassifier instance is used for classification.
|
172 |
|
173 |
The model has been trained using an efficient few-shot learning technique that involves:
|
174 |
|
|
|
180 |
### Model Description
|
181 |
- **Model Type:** SetFit
|
182 |
- **Sentence Transformer body:** [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0)
|
183 |
+
- **Classification head:** a OneVsRestClassifier instance
|
184 |
- **Maximum Sequence Length:** 512 tokens
|
185 |
+
- **Number of Classes:** 13 classes
|
186 |
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
187 |
<!-- - **Language:** Unknown -->
|
188 |
<!-- - **License:** Unknown -->
|
|
|
193 |
- **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
|
194 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
195 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
196 |
## Uses
|
197 |
|
198 |
### Direct Use for Inference
|
config_setfit.json
CHANGED
@@ -1,4 +1,18 @@
|
|
1 |
{
|
2 |
-
"labels":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
"normalize_embeddings": false
|
4 |
}
|
|
|
1 |
{
|
2 |
+
"labels": [
|
3 |
+
"Baking",
|
4 |
+
"Grilling",
|
5 |
+
"Slow cooking",
|
6 |
+
"Pressure cooking",
|
7 |
+
"Air-fryer",
|
8 |
+
"Smoking",
|
9 |
+
"Pickling",
|
10 |
+
"Roasting",
|
11 |
+
"Steaming",
|
12 |
+
"Boiling",
|
13 |
+
"Saut\u00e9ing",
|
14 |
+
"Sous vide",
|
15 |
+
"Stir-frying"
|
16 |
+
],
|
17 |
"normalize_embeddings": false
|
18 |
}
|
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:53bcc40c8e1e047e7d2e334879f57555a428fff6e22cb405cd9b3ec44032b09f
|
3 |
+
size 45316
|