Hugging Face's logo

weizhou03
/
nomic-embed-text-v1.5

Text Classification
setfit
Safetensors
GGUF
sentence-transformers
nomic_bert
generated_from_setfit_trainer
custom_code
Eval Results
feature-extraction
Model card Files
xet
Community
weizhou03 commited on
Commit
61bbddb
·
verified ·
1 Parent(s): e848f79

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +142 -168
  2. model.safetensors +1 -1
  3. model_head.pkl +1 -1
README.md CHANGED
@@ -5,11 +5,11 @@ tags:
5
  - text-classification
6
  - generated_from_setfit_trainer
7
  widget:
8
- - text: Add a navigation menu text
9
- - text: Change to square format
10
- - text: Change the button text
11
- - text: Make the object stand alone
12
- - text: Mirror the logo vertically
13
  metrics:
14
  - accuracy
15
  pipeline_tag: text-classification
@@ -28,7 +28,7 @@ model-index:
28
  split: test
29
  metrics:
30
  - type: accuracy
31
- value: 0.5353535353535354
32
  name: Accuracy
33
  ---
34
 
@@ -60,78 +60,78 @@ The model has been trained using an efficient few-shot learning technique that i
60
  - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
61
 
62
  ### Model Labels
63
- | Label | Examples |
64
- |:------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
65
- | 0 | <ul><li>'Add a corporate presentation background'</li><li>'Insert a modern icon set for the design'</li><li>'Add a mountain landscape background to the page'</li></ul> |
66
- | 1 | <ul><li>'Find me some shape options for this design'</li><li>'I need some professional-looking assets'</li><li>'Can you recommend some images that would work well here?'</li></ul> |
67
- | 2 | <ul><li>'Add a date and time for the event'</li><li>'Insert a disclaimer text'</li><li>'Add a navigation menu text'</li></ul> |
68
- | 3 | <ul><li>'Distribute the icons evenly'</li><li>'Align all the text elements to the left'</li><li>'Align the footer elements'</li></ul> |
69
- | 4 | <ul><li>'Make the button pulse'</li><li>'Add a flip animation'</li><li>'Make the text glow'</li></ul> |
70
- | 5 | <ul><li>'Make everything fade in gradually'</li><li>'Make the page bounce in from the top'</li><li>'Add a spiral animation to the page'</li></ul> |
71
- | 6 | <ul><li>'Change the building color'</li><li>'Remove the unwanted text overlay'</li><li>'Add a party hat to the dog'</li></ul> |
72
- | 7 | <ul><li>'Remove the draft image'</li><li>'Delete the backup copy'</li><li>'Remove the unwanted image'</li></ul> |
73
- | 8 | <ul><li>'What image editing tools do you have?'</li><li>'How do I create a template?'</li><li>'How can I align elements properly?'</li></ul> |
74
- | 9 | <ul><li>'Distribute the buttons around the center image'</li><li>'Place the elements in a circular arrangement'</li><li>'Arrange the images in a circular layout'</li></ul> |
75
- | 10 | <ul><li>'Duplicate the design'</li><li>'Make a second version'</li><li>'Create a new version'</li></ul> |
76
- | 11 | <ul><li>'Duplicate the icon and move it'</li><li>'Duplicate the text and align it differently'</li><li>'Copy the image and apply a filter'</li></ul> |
77
- | 12 | <ul><li>'Duplicate the logo to page 3'</li><li>'Copy the text to the last page'</li><li>'Copy the navigation to the next page'</li></ul> |
78
- | 13 | <ul><li>'Fix the typographic errors'</li><li>'Improve the text flow'</li><li>'Fix the font consistency'</li></ul> |
79
- | 14 | <ul><li>'Mirror the icon horizontally'</li><li>'Mirror the logo vertically'</li><li>'Flip the image horizontally'</li></ul> |
80
- | 15 | <ul><li>'Create a photo of a futuristic city'</li><li>'Generate a picture of a tropical beach'</li><li>'Generate a picture of a cat playing with yarn'</li></ul> |
81
- | 16 | <ul><li>'Create a card for a birthday party'</li><li>'Create a flyer for a happy birthday party'</li><li>'Generate an Instagram post for a birthday'</li></ul> |
82
- | 17 | <ul><li>'Group the navigation elements'</li><li>'Combine the shape and text'</li><li>'Combine the image and overlay'</li></ul> |
83
- | 18 | <ul><li>'Move the shape to the bottom'</li><li>'Position the icon at (0, 0)'</li><li>'Move the shape to coordinates (100, 100)'</li></ul> |
84
- | 19 | <ul><li>'Add a sepia tone effect'</li><li>'Apply a modern filter'</li><li>'Make the image black and white'</li></ul> |
85
- | 20 | <ul><li>'Suggest some shape designs'</li><li>'Show me pattern options'</li><li>'Find me border designs'</li></ul> |
86
- | 21 | <ul><li>'Restore the previous opacity'</li><li>'Redo the text edit'</li><li>'Restore the previous color'</li></ul> |
87
- | 22 | <ul><li>'Make the image have no background'</li><li>'Remove the background from the animal'</li><li>'Remove the background from the item'</li></ul> |
88
- | 23 | <ul><li>'Delete the unwanted text'</li><li>'Remove the person from the background'</li><li>'Remove the graffiti'</li></ul> |
89
- | 24 | <ul><li>'Replace the illustration'</li><li>'Change the background image'</li><li>'Change the product photo'</li></ul> |
90
- | 25 | <ul><li>'Change the button text'</li><li>'Update the navigation text'</li><li>'Update the description'</li></ul> |
91
- | 26 | <ul><li>'Remove all modifications'</li><li>'Restore the original colors'</li><li>'Remove all effects from the image'</li></ul> |
92
- | 27 | <ul><li>'Scale the text up'</li><li>'Make the shape smaller'</li><li>'Reduce the shape size'</li></ul> |
93
- | 28 | <ul><li>'Change to square format'</li><li>'Change to poster size'</li><li>'Make the page smaller'</li></ul> |
94
- | 29 | <ul><li>'Rotate the text 45 degrees'</li><li>'Rotate the text 15 degrees'</li><li>'Turn the image counterclockwise'</li></ul> |
95
- | 30 | <ul><li>'Distribute the shapes randomly'</li><li>'Distribute the leaves randomly on the page'</li><li>'Scatter the confetti around the design'</li></ul> |
96
- | 31 | <ul><li>'Select the sidebar elements'</li><li>'Select the background image'</li><li>'Select the footer content'</li></ul> |
97
- | 32 | <ul><li>'Change to a dark background'</li><li>'Make the background transparent'</li><li>'Change to a neutral background'</li></ul> |
98
- | 33 | <ul><li>'Set the blend mode to soft light'</li><li>'Set the blend mode to darken'</li><li>'Set the blend mode to normal'</li></ul> |
99
- | 34 | <ul><li>'Add a depth of field blur'</li><li>'Add a directional blur'</li><li>'Add a soft blur effect'</li></ul> |
100
- | 35 | <ul><li>'Change the border style to solid'</li><li>'Change the border to dotted'</li><li>'Add a double border to the image'</li></ul> |
101
- | 36 | <ul><li>'Brighten the shadows'</li><li>'Make the image brighter'</li><li>'Brighten the highlights'</li></ul> |
102
- | 37 | <ul><li>'Bring the title to the front'</li><li>'Bring the shape to the front'</li><li>'Move the logo to the top layer'</li></ul> |
103
- | 38 | <ul><li>'Make the image more intense'</li><li>'Enhance the contrast ratio'</li><li>'Increase the tonal range'</li></ul> |
104
- | 39 | <ul><li>'Make the image round'</li><li>'Make the image rectangular'</li><li>'Crop to a heart shape'</li></ul> |
105
- | 40 | <ul><li>'Add a sharp drop shadow'</li><li>'Add a gradient shadow'</li><li>'Add a hard shadow edge'</li></ul> |
106
- | 41 | <ul><li>'Fill the shape with orange'</li><li>'Fill the element with brown'</li><li>'Change the shape color to blue'</li></ul> |
107
- | 42 | <ul><li>'Make the text more readable'</li><li>'Increase the heading size'</li><li>'Make the description smaller'</li></ul> |
108
- | 43 | <ul><li>'Make the text bold and italic'</li><li>'Add strikethrough to the text'</li><li>'Add bold to the title'</li></ul> |
109
- | 44 | <ul><li>'Use a contemporary font'</li><li>'Use a professional font'</li><li>'Use a serif font for the heading'</li></ul> |
110
- | 45 | <ul><li>'Make the bright areas brighter'</li><li>'Make the highlights more prominent'</li><li>'Enhance the bright spots'</li></ul> |
111
- | 46 | <ul><li>'Set the image as background layer'</li><li>'Set the picture as background fill'</li><li>'Make the photo cover the background'</li></ul> |
112
- | 47 | <ul><li>'Reduce the character spacing'</li><li>'Add letter spacing to the logo'</li><li>'Increase spacing between characters'</li></ul> |
113
- | 48 | <ul><li>'Make the lines tighter'</li><li>'Spread out the text lines'</li><li>'Increase the paragraph spacing'</li></ul> |
114
- | 49 | <ul><li>'Reduce the opacity of the overlay'</li><li>'Increase the transparency of the image'</li><li>'Make the shape more opaque'</li></ul> |
115
- | 50 | <ul><li>'Make the paragraphs closer together'</li><li>'Increase the text block spacing'</li><li>'Reduce the paragraph spacing'</li></ul> |
116
- | 51 | <ul><li>'Make the colors more intense'</li><li>'Increase the color depth'</li><li>'Make the image more colorful'</li></ul> |
117
- | 52 | <ul><li>'Darken the shadows in the image'</li><li>'Increase the shadow intensity'</li><li>'Enhance the shadow depth'</li></ul> |
118
- | 53 | <ul><li>'Enhance the image clarity'</li><li>'Increase the image sharpness'</li><li>'Sharpen the image details'</li></ul> |
119
- | 54 | <ul><li>'Center the button text'</li><li>'Justify the paragraph text'</li><li>'Center the title text'</li></ul> |
120
- | 55 | <ul><li>'Create a border around the text'</li><li>'Add a glow effect behind the text'</li><li>'Add a colored background to the text'</li></ul> |
121
- | 56 | <ul><li>'Create text in a radial pattern'</li><li>'Create text that follows a circle'</li><li>'Create text that follows a shape'</li></ul> |
122
- | 57 | <ul><li>'Convert to a bulleted list'</li><li>'Make the text into bullet points'</li><li>'Make the text into a list with bullets'</li></ul> |
123
- | 58 | <ul><li>'Create a soft shadow behind the text'</li><li>'Add a dramatic shadow effect'</li><li>'Create a shadow for the text'</li></ul> |
124
- | 59 | <ul><li>'Add warm undertones to the photo'</li><li>'Add warm color grading'</li><li>'Make the photo more golden hour'</li></ul> |
125
- | 60 | <ul><li>'I need to add my own image'</li><li>'Open the image upload tool'</li><li>'Show me how to upload files'</li></ul> |
126
- | 61 | <ul><li>'Revert the color change'</li><li>'Undo the last modification'</li><li>'Undo the text edit'</li></ul> |
127
- | 62 | <ul><li>'Separate the grouped components'</li><li>'Ungroup the combined elements'</li><li>'Break up the grouped objects'</li></ul> |
128
 
129
  ## Evaluation
130
 
131
  ### Metrics
132
  | Label | Accuracy |
133
  |:--------|:---------|
134
- | **all** | 0.5354 |
135
 
136
  ## Uses
137
 
@@ -151,7 +151,7 @@ from setfit import SetFitModel
151
  # Download from the 🤗 Hub
152
  model = SetFitModel.from_pretrained("setfit_model_id")
153
  # Run inference
154
- preds = model("Change the button text")
155
  ```
156
 
157
  <!--
@@ -183,73 +183,73 @@ preds = model("Change the button text")
183
  ### Training Set Metrics
184
  | Training set | Min | Median | Max |
185
  |:-------------|:----|:-------|:----|
186
- | Word count | 3 | 5.1778 | 10 |
187
 
188
  | Label | Training Sample Count |
189
  |:------|:----------------------|
190
- | 0 | 5 |
191
- | 1 | 5 |
192
- | 2 | 5 |
193
- | 3 | 5 |
194
- | 4 | 5 |
195
- | 5 | 5 |
196
- | 6 | 5 |
197
- | 7 | 5 |
198
- | 8 | 5 |
199
- | 9 | 5 |
200
- | 10 | 5 |
201
- | 11 | 5 |
202
- | 12 | 5 |
203
- | 13 | 5 |
204
- | 14 | 5 |
205
- | 15 | 5 |
206
- | 16 | 5 |
207
- | 17 | 5 |
208
- | 18 | 5 |
209
- | 19 | 5 |
210
- | 20 | 5 |
211
- | 21 | 5 |
212
- | 22 | 5 |
213
- | 23 | 5 |
214
- | 24 | 5 |
215
- | 25 | 5 |
216
- | 26 | 5 |
217
- | 27 | 5 |
218
- | 28 | 5 |
219
- | 29 | 5 |
220
- | 30 | 5 |
221
- | 31 | 5 |
222
- | 32 | 5 |
223
- | 33 | 5 |
224
- | 34 | 5 |
225
- | 35 | 5 |
226
- | 36 | 5 |
227
- | 37 | 5 |
228
- | 38 | 5 |
229
- | 39 | 5 |
230
- | 40 | 5 |
231
- | 41 | 5 |
232
- | 42 | 5 |
233
- | 43 | 5 |
234
- | 44 | 5 |
235
- | 45 | 5 |
236
- | 46 | 5 |
237
- | 47 | 5 |
238
- | 48 | 5 |
239
- | 49 | 5 |
240
- | 50 | 5 |
241
- | 51 | 5 |
242
- | 52 | 5 |
243
- | 53 | 5 |
244
- | 54 | 5 |
245
- | 55 | 5 |
246
- | 56 | 5 |
247
- | 57 | 5 |
248
- | 58 | 5 |
249
- | 59 | 5 |
250
- | 60 | 5 |
251
- | 61 | 5 |
252
- | 62 | 5 |
253
 
254
  ### Training Hyperparameters
255
  - batch_size: (64, 64)
@@ -272,37 +272,11 @@ preds = model("Change the button text")
272
  ### Training Results
273
  | Epoch | Step | Training Loss | Validation Loss |
274
  |:------:|:----:|:-------------:|:---------------:|
275
- | 0.0007 | 1 | 0.1545 | - |
276
- | 0.0328 | 50 | 0.1187 | - |
277
- | 0.0655 | 100 | 0.0521 | - |
278
- | 0.0983 | 150 | 0.0208 | - |
279
- | 0.1311 | 200 | 0.0123 | - |
280
- | 0.1638 | 250 | 0.0096 | - |
281
- | 0.1966 | 300 | 0.0056 | - |
282
- | 0.2294 | 350 | 0.0036 | - |
283
- | 0.2621 | 400 | 0.0027 | - |
284
- | 0.2949 | 450 | 0.0017 | - |
285
- | 0.3277 | 500 | 0.0007 | - |
286
- | 0.3604 | 550 | 0.0009 | - |
287
- | 0.3932 | 600 | 0.0009 | - |
288
- | 0.4260 | 650 | 0.0003 | - |
289
- | 0.4587 | 700 | 0.0003 | - |
290
- | 0.4915 | 750 | 0.0004 | - |
291
- | 0.5242 | 800 | 0.0004 | - |
292
- | 0.5570 | 850 | 0.0002 | - |
293
- | 0.5898 | 900 | 0.0001 | - |
294
- | 0.6225 | 950 | 0.0001 | - |
295
- | 0.6553 | 1000 | 0.0001 | - |
296
- | 0.6881 | 1050 | 0.0001 | - |
297
- | 0.7208 | 1100 | 0.0001 | - |
298
- | 0.7536 | 1150 | 0.0001 | - |
299
- | 0.7864 | 1200 | 0.0001 | - |
300
- | 0.8191 | 1250 | 0.0001 | - |
301
- | 0.8519 | 1300 | 0.0001 | - |
302
- | 0.8847 | 1350 | 0.0001 | - |
303
- | 0.9174 | 1400 | 0.0001 | - |
304
- | 0.9502 | 1450 | 0.0001 | - |
305
- | 0.9830 | 1500 | 0.0001 | - |
306
 
307
  ### Framework Versions
308
  - Python: 3.12.11
 
5
  - text-classification
6
  - generated_from_setfit_trainer
7
  widget:
8
+ - text: I need some modern design assets
9
+ - text: Reduce the letter spacing
10
+ - text: Group the menu items
11
+ - text: Create a photo of a delicious pizza
12
+ - text: Mirror the icon horizontally
13
  metrics:
14
  - accuracy
15
  pipeline_tag: text-classification
 
28
  split: test
29
  metrics:
30
  - type: accuracy
31
+ value: 0.4163860830527497
32
  name: Accuracy
33
  ---
34
 
 
60
  - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
61
 
62
  ### Model Labels
63
+ | Label | Examples |
64
+ |:------|:--------------------------------------------------------------------------------------------------------------------|
65
+ | 0 | <ul><li>'Add a corporate presentation background'</li><li>'Insert a modern icon set for the design'</li></ul> |
66
+ | 1 | <ul><li>'Show me some background options for my design'</li><li>'I need some modern design assets'</li></ul> |
67
+ | 2 | <ul><li>'Add a footer with contact information'</li><li>'Insert a descriptive caption for the image'</li></ul> |
68
+ | 3 | <ul><li>'Align the shapes to the top'</li><li>'Right align the buttons'</li></ul> |
69
+ | 4 | <ul><li>'Make the image slide from left'</li><li>'Make the image bounce in'</li></ul> |
70
+ | 5 | <ul><li>'Add a spiral animation to the page'</li><li>'Make the page bounce in from the top'</li></ul> |
71
+ | 6 | <ul><li>'Take out the cat and put a dog in the image'</li><li>"Add glasses to the person's face"</li></ul> |
72
+ | 7 | <ul><li>'Delete the test element'</li><li>'Remove the unused icon'</li></ul> |
73
+ | 8 | <ul><li>'How do I change the font size?'</li><li>'How can I duplicate elements?'</li></ul> |
74
+ | 9 | <ul><li>'Distribute the buttons around the center image'</li><li>'Arrange the shapes in a circle pattern'</li></ul> |
75
+ | 10 | <ul><li>'Create a variation of this design'</li><li>'Make a copy for editing'</li></ul> |
76
+ | 11 | <ul><li>'Duplicate the shape and change its color'</li><li>'Copy the image and make it smaller'</li></ul> |
77
+ | 12 | <ul><li>'Copy the text to the last page'</li><li>'Duplicate the footer to page 2'</li></ul> |
78
+ | 13 | <ul><li>'Fix the text layout'</li><li>'Improve the overall typography'</li></ul> |
79
+ | 14 | <ul><li>'Mirror the icon horizontally'</li><li>'Flip the shape vertically'</li></ul> |
80
+ | 15 | <ul><li>'Create a photo of a beautiful garden'</li><li>'Create a photo of a delicious pizza'</li></ul> |
81
+ | 16 | <ul><li>'Generate an Instagram post for a birthday'</li><li>'Generate a Facebook post for a birthday'</li></ul> |
82
+ | 17 | <ul><li>'Group the icons together'</li><li>'Group the menu items'</li></ul> |
83
+ | 18 | <ul><li>'Position the image at the top right'</li><li>'Move the image to the center'</li></ul> |
84
+ | 19 | <ul><li>'Add a dreamy effect'</li><li>'Make the image pop with color'</li></ul> |
85
+ | 20 | <ul><li>'Suggest some shape designs'</li><li>'Suggest some button designs'</li></ul> |
86
+ | 21 | <ul><li>'Redo the filter effect'</li><li>'Bring back the original position'</li></ul> |
87
+ | 22 | <ul><li>'Remove the background from the icon'</li><li>'Remove the background from the logo'</li></ul> |
88
+ | 23 | <ul><li>'Delete the watermark'</li><li>'Remove the graffiti'</li></ul> |
89
+ | 24 | <ul><li>'Change the decorative element'</li><li>'Change the background image'</li></ul> |
90
+ | 25 | <ul><li>'Update the body text'</li><li>'Change the subtitle'</li></ul> |
91
+ | 26 | <ul><li>'Remove all filters from the image'</li><li>'Restore the original brightness'</li></ul> |
92
+ | 27 | <ul><li>'Reduce the shape size'</li><li>'Increase the image size'</li></ul> |
93
+ | 28 | <ul><li>'Change to poster size'</li><li>'Resize to Facebook post dimensions'</li></ul> |
94
+ | 29 | <ul><li>'Rotate the element 60 degrees'</li><li>'Rotate the image 90 degrees'</li></ul> |
95
+ | 30 | <ul><li>'Randomly place the icons on the page'</li><li>'Randomly place the shapes on the page'</li></ul> |
96
+ | 31 | <ul><li>'Select the decorative elements'</li><li>'Choose the heading text'</li></ul> |
97
+ | 32 | <ul><li>'Make the background transparent'</li><li>'Change to a dark background'</li></ul> |
98
+ | 33 | <ul><li>'Change the blend mode to screen'</li><li>'Set the blend mode to darken'</li></ul> |
99
+ | 34 | <ul><li>'Add a gaussian blur'</li><li>'Blur the background elements'</li></ul> |
100
+ | 35 | <ul><li>'Add a gradient border'</li><li>'Change the border opacity'</li></ul> |
101
+ | 36 | <ul><li>'Make the image more brilliant'</li><li>'Increase the brightness contrast'</li></ul> |
102
+ | 37 | <ul><li>'Send the background to the back'</li><li>'Send the watermark to the back'</li></ul> |
103
+ | 38 | <ul><li>'Make the image more dramatic'</li><li>'Make the image more defined'</li></ul> |
104
+ | 39 | <ul><li>'Make the image square'</li><li>'Crop the image to a circle'</li></ul> |
105
+ | 40 | <ul><li>'Add a directional shadow'</li><li>'Create a floating shadow'</li></ul> |
106
+ | 41 | <ul><li>'Change the color to silver'</li><li>'Change the button color to purple'</li></ul> |
107
+ | 42 | <ul><li>'Increase the body text size'</li><li>'Increase the heading size'</li></ul> |
108
+ | 43 | <ul><li>'Add strikethrough to the text'</li><li>'Add strikethrough to the label'</li></ul> |
109
+ | 44 | <ul><li>'Use a serif font for the heading'</li><li>'Change to a clean font'</li></ul> |
110
+ | 45 | <ul><li>'Brighten the light areas'</li><li>'Enhance the bright spots'</li></ul> |
111
+ | 46 | <ul><li>'Set the image as full-page background'</li><li>'Make the image the background layer'</li></ul> |
112
+ | 47 | <ul><li>'Reduce the letter spacing'</li><li>'Add more space between letters'</li></ul> |
113
+ | 48 | <ul><li>'Make the text more compact'</li><li>'Reduce the line height'</li></ul> |
114
+ | 49 | <ul><li>'Increase the transparency of the element'</li><li>'Make the element semi-transparent'</li></ul> |
115
+ | 50 | <ul><li>'Add spacing between text blocks'</li><li>'Make the paragraphs more compact'</li></ul> |
116
+ | 51 | <ul><li>'Reduce the color richness'</li><li>'Reduce the color vibrancy'</li></ul> |
117
+ | 52 | <ul><li>'Make the shadows more prominent'</li><li>'Increase the shadow intensity'</li></ul> |
118
+ | 53 | <ul><li>'Increase the image resolution'</li><li>'Make the details more sharp'</li></ul> |
119
+ | 54 | <ul><li>'Right align the caption'</li><li>'Center the button text'</li></ul> |
120
+ | 55 | <ul><li>'Create a decorative border'</li><li>'Create a border around the text'</li></ul> |
121
+ | 56 | <ul><li>'Make the text flow in an arc'</li><li>'Create text that follows a circle'</li></ul> |
122
+ | 57 | <ul><li>'Convert the text to a list format'</li><li>'Make the text into a numbered sequence'</li></ul> |
123
+ | 58 | <ul><li>'Add a dramatic text shadow'</li><li>'Add a colored glow to the text'</li></ul> |
124
+ | 59 | <ul><li>'Make the image more amber'</li><li>'Increase the warm color balance'</li></ul> |
125
+ | 60 | <ul><li>'Open the file upload dialog'</li><li>'I need to upload a logo'</li></ul> |
126
+ | 61 | <ul><li>'Go back to the original position'</li><li>'Go back to the original text'</li></ul> |
127
+ | 62 | <ul><li>'Separate the combined elements'</li><li>'Ungroup the merged elements'</li></ul> |
128
 
129
  ## Evaluation
130
 
131
  ### Metrics
132
  | Label | Accuracy |
133
  |:--------|:---------|
134
+ | **all** | 0.4164 |
135
 
136
  ## Uses
137
 
 
151
  # Download from the 🤗 Hub
152
  model = SetFitModel.from_pretrained("setfit_model_id")
153
  # Run inference
154
+ preds = model("Group the menu items")
155
  ```
156
 
157
  <!--
 
183
  ### Training Set Metrics
184
  | Training set | Min | Median | Max |
185
  |:-------------|:----|:-------|:----|
186
+ | Word count | 3 | 5.2302 | 11 |
187
 
188
  | Label | Training Sample Count |
189
  |:------|:----------------------|
190
+ | 0 | 2 |
191
+ | 1 | 2 |
192
+ | 2 | 2 |
193
+ | 3 | 2 |
194
+ | 4 | 2 |
195
+ | 5 | 2 |
196
+ | 6 | 2 |
197
+ | 7 | 2 |
198
+ | 8 | 2 |
199
+ | 9 | 2 |
200
+ | 10 | 2 |
201
+ | 11 | 2 |
202
+ | 12 | 2 |
203
+ | 13 | 2 |
204
+ | 14 | 2 |
205
+ | 15 | 2 |
206
+ | 16 | 2 |
207
+ | 17 | 2 |
208
+ | 18 | 2 |
209
+ | 19 | 2 |
210
+ | 20 | 2 |
211
+ | 21 | 2 |
212
+ | 22 | 2 |
213
+ | 23 | 2 |
214
+ | 24 | 2 |
215
+ | 25 | 2 |
216
+ | 26 | 2 |
217
+ | 27 | 2 |
218
+ | 28 | 2 |
219
+ | 29 | 2 |
220
+ | 30 | 2 |
221
+ | 31 | 2 |
222
+ | 32 | 2 |
223
+ | 33 | 2 |
224
+ | 34 | 2 |
225
+ | 35 | 2 |
226
+ | 36 | 2 |
227
+ | 37 | 2 |
228
+ | 38 | 2 |
229
+ | 39 | 2 |
230
+ | 40 | 2 |
231
+ | 41 | 2 |
232
+ | 42 | 2 |
233
+ | 43 | 2 |
234
+ | 44 | 2 |
235
+ | 45 | 2 |
236
+ | 46 | 2 |
237
+ | 47 | 2 |
238
+ | 48 | 2 |
239
+ | 49 | 2 |
240
+ | 50 | 2 |
241
+ | 51 | 2 |
242
+ | 52 | 2 |
243
+ | 53 | 2 |
244
+ | 54 | 2 |
245
+ | 55 | 2 |
246
+ | 56 | 2 |
247
+ | 57 | 2 |
248
+ | 58 | 2 |
249
+ | 59 | 2 |
250
+ | 60 | 2 |
251
+ | 61 | 2 |
252
+ | 62 | 2 |
253
 
254
  ### Training Hyperparameters
255
  - batch_size: (64, 64)
 
272
  ### Training Results
273
  | Epoch | Step | Training Loss | Validation Loss |
274
  |:------:|:----:|:-------------:|:---------------:|
275
+ | 0.0041 | 1 | 0.1097 | - |
276
+ | 0.2041 | 50 | 0.0477 | - |
277
+ | 0.4082 | 100 | 0.0106 | - |
278
+ | 0.6122 | 150 | 0.0045 | - |
279
+ | 0.8163 | 200 | 0.0032 | - |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
280
 
281
  ### Framework Versions
282
  - Python: 3.12.11
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fffecbd38859dda02c39c881c1dd347654613c58bb22dcd705706dd3c9247709
3
  size 546938168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8dedfd91f01deae16c4face0912355b8847fb4c7755c8cedd6ec846f31ad137d
3
  size 546938168
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:963cbad21c41b69e67104f59ae67989240513569fb9d1218df64a659c819609a
3
  size 388927
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f65355373e7c0f90fb3b5584001fca42e18d7f176f7fa0c9e0c7e0797dd887d4
3
  size 388927