Text Classification
setfit
Safetensors
sentence-transformers
bert
generated_from_setfit_trainer
Eval Results (legacy)
text-embeddings-inference
Instructions to use akhooli/setfit_ar_sst2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- setfit
How to use akhooli/setfit_ar_sst2 with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("akhooli/setfit_ar_sst2") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds) - sentence-transformers
How to use akhooli/setfit_ar_sst2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("akhooli/setfit_ar_sst2") 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
Push model using huggingface_hub.
Browse files- README.md +34 -22
- config_setfit.json +2 -2
- model.safetensors +1 -1
- model_head.pkl +1 -1
README.md
CHANGED
|
@@ -10,14 +10,15 @@ tags:
|
|
| 10 |
- text-classification
|
| 11 |
- generated_from_setfit_trainer
|
| 12 |
widget:
|
| 13 |
-
- text: '
|
| 14 |
-
- text: '
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
- text: '
|
| 19 |
-
و
|
| 20 |
-
|
|
|
|
| 21 |
inference: true
|
| 22 |
model-index:
|
| 23 |
- name: SetFit with akhooli/sbert_ar_nli_500k_norm
|
|
@@ -31,7 +32,7 @@ model-index:
|
|
| 31 |
split: test
|
| 32 |
metrics:
|
| 33 |
- type: accuracy
|
| 34 |
-
value: 0.
|
| 35 |
name: Accuracy
|
| 36 |
---
|
| 37 |
|
|
@@ -63,19 +64,19 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
| 63 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
| 64 |
|
| 65 |
### Model Labels
|
| 66 |
-
| Label | Examples
|
| 67 |
-
|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
|
| 68 |
-
|
|
| 69 |
-
|
|
| 70 |
-
| 1 | <ul><li>'
|
| 71 |
-
| 0 | <ul><li>'
|
| 72 |
|
| 73 |
## Evaluation
|
| 74 |
|
| 75 |
### Metrics
|
| 76 |
| Label | Accuracy |
|
| 77 |
|:--------|:---------|
|
| 78 |
-
| **all** | 0.
|
| 79 |
|
| 80 |
## Uses
|
| 81 |
|
|
@@ -95,7 +96,7 @@ from setfit import SetFitModel
|
|
| 95 |
# Download from the 🤗 Hub
|
| 96 |
model = SetFitModel.from_pretrained("akhooli/setfit")
|
| 97 |
# Run inference
|
| 98 |
-
preds = model("
|
| 99 |
```
|
| 100 |
|
| 101 |
<!--
|
|
@@ -127,7 +128,7 @@ preds = model("ليس بالضبط ركب النحل ")
|
|
| 127 |
### Training Set Metrics
|
| 128 |
| Training set | Min | Median | Max |
|
| 129 |
|:-------------|:----|:--------|:----|
|
| 130 |
-
| Word count | 5 | 16.
|
| 131 |
|
| 132 |
| Label | Training Sample Count |
|
| 133 |
|:---------|:----------------------|
|
|
@@ -153,10 +154,21 @@ preds = model("ليس بالضبط ركب النحل ")
|
|
| 153 |
- load_best_model_at_end: False
|
| 154 |
|
| 155 |
### Training Results
|
| 156 |
-
| Epoch
|
| 157 |
-
|:-----:|:----:|:-------------:|:---------------:|
|
| 158 |
-
| 0.
|
| 159 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 160 |
|
| 161 |
### Framework Versions
|
| 162 |
- Python: 3.10.14
|
|
|
|
| 10 |
- text-classification
|
| 11 |
- generated_from_setfit_trainer
|
| 12 |
widget:
|
| 13 |
+
- text: 'مضحك بشكل غامق وثاقب في كثير من الأحيان. '
|
| 14 |
+
- text: 'على عكس معظم الرسوم المتحركة، التي يبدو أن معظم معجبيها المتحمسين خارج اليابان
|
| 15 |
+
هم شباب انطوائيون لديهم هوس خيالي، لا تبدو متروبوليس أبدًا حدثًا ميؤوسًا منه. '
|
| 16 |
+
- text: فيلم فظيع تمامًا - غبي، وفوضوي من حيث السرد، وقذر بصريًا... مزيج غريب من "الشيء"
|
| 17 |
+
وصرخة الشيخوخة. '$$ 0
|
| 18 |
+
- text: 'الفيلم عبارة عن تقاطع فجر الموتى مع أشباح المريخ للمخرج جون كاربنتر، مع كائنات
|
| 19 |
+
زومبي ليست مخيفة كالأولى وقطارات ليست كبيرة كالثانية. '
|
| 20 |
+
- text: 'كوميديا رومانسية نعم، لكن مع شخصيات تفكر وتتحدث عن أهدافها، وتعمل على اتخاذ
|
| 21 |
+
قرارات صعبة. '
|
| 22 |
inference: true
|
| 23 |
model-index:
|
| 24 |
- name: SetFit with akhooli/sbert_ar_nli_500k_norm
|
|
|
|
| 32 |
split: test
|
| 33 |
metrics:
|
| 34 |
- type: accuracy
|
| 35 |
+
value: 0.77
|
| 36 |
name: Accuracy
|
| 37 |
---
|
| 38 |
|
|
|
|
| 64 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
| 65 |
|
| 66 |
### Model Labels
|
| 67 |
+
| Label | Examples |
|
| 68 |
+
|:------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 69 |
+
| | <ul><li>"إن مور يشبه الثور التقدمي في متجر للخزف الصيني، وهو محرض يقتحم الأفكار ومجموعات المصالح الخاصة بينما يجمع علامته التجارية الخاصة من الليبرالية. '$$ 0"</li><li>'ارتكب الكاتب والمخرج ستيفن جاغان خطأً فادحًا لكونه ما يسميه الإنجليز "ذكيًا جدًا بمقدار النصف". \'$$ 0'</li><li>'يحمل هذا الفيلم الرقيق على كتفيه الرشيقتين، ويخوض تشان في الكتابة الفاسدة والاتجاه والتوقيت بابتسامة تقول: "إذا بقيت إيجابيًا، فربما أستطيع أن أعرض واحدة من أعظم صوري، المعلم السكير". \'$$ 0'</li></ul> |
|
| 70 |
+
| 0 | <ul><li>'بناء زائف يعتمد على النظرية وخفة اليد والفرضيات الخاطئة. '</li><li>'يحفر كتاب السيناريو أنفسهم بشكل أعمق في كل مرة يرمون فيها المنطق والعلم إلى ما هو في الأساس خيال "زنزانات وتنانين" بأسلحة عسكرية حديثة ... '</li><li>'يعاني من نص مسطح وميزانية منخفضة. '</li></ul> |
|
| 71 |
+
| 1 | <ul><li>'يملأ رامزي ومورتون دراسة الشخصية هذه بالقوة الشعرية والشعور المزدهر. '</li><li>'مضحك بشكل غامق وثاقب في كثير من الأحيان. '</li><li>"-lrb- `` اعتني بقطتي '' -rrb- هو فيلم صغير لطيف بصراحة يأخذنا في فحص حياة الشباب البالغين في المناطق الحضرية في كوريا الجنوبية من خلال قلوب وعقول المديرين الخمسة. "</li></ul> |
|
| 72 |
+
| 0 | <ul><li>'أنا أكره هذا الفيلم '</li><li>'ليس بالضبط ركب النحل '</li></ul> |
|
| 73 |
|
| 74 |
## Evaluation
|
| 75 |
|
| 76 |
### Metrics
|
| 77 |
| Label | Accuracy |
|
| 78 |
|:--------|:---------|
|
| 79 |
+
| **all** | 0.77 |
|
| 80 |
|
| 81 |
## Uses
|
| 82 |
|
|
|
|
| 96 |
# Download from the 🤗 Hub
|
| 97 |
model = SetFitModel.from_pretrained("akhooli/setfit")
|
| 98 |
# Run inference
|
| 99 |
+
preds = model("مضحك بشكل غامق وثاقب في كثير من الأحيان. ")
|
| 100 |
```
|
| 101 |
|
| 102 |
<!--
|
|
|
|
| 128 |
### Training Set Metrics
|
| 129 |
| Training set | Min | Median | Max |
|
| 130 |
|:-------------|:----|:--------|:----|
|
| 131 |
+
| Word count | 5 | 16.7013 | 31 |
|
| 132 |
|
| 133 |
| Label | Training Sample Count |
|
| 134 |
|:---------|:----------------------|
|
|
|
|
| 154 |
- load_best_model_at_end: False
|
| 155 |
|
| 156 |
### Training Results
|
| 157 |
+
| Epoch | Step | Training Loss | Validation Loss |
|
| 158 |
+
|:------:|:----:|:-------------:|:---------------:|
|
| 159 |
+
| 0.0156 | 1 | 0.3152 | - |
|
| 160 |
+
| 0.7812 | 50 | 0.1993 | - |
|
| 161 |
+
| 1.5625 | 100 | 0.031 | - |
|
| 162 |
+
| 2.3438 | 150 | 0.0012 | - |
|
| 163 |
+
| 3.125 | 200 | 0.0007 | - |
|
| 164 |
+
| 3.9062 | 250 | 0.0006 | - |
|
| 165 |
+
| 4.6875 | 300 | 0.0003 | - |
|
| 166 |
+
| 5.4688 | 350 | 0.0004 | - |
|
| 167 |
+
| 6.25 | 400 | 0.0003 | - |
|
| 168 |
+
| 7.0312 | 450 | 0.0003 | - |
|
| 169 |
+
| 7.8125 | 500 | 0.0003 | - |
|
| 170 |
+
| 8.5938 | 550 | 0.0003 | - |
|
| 171 |
+
| 9.375 | 600 | 0.0002 | - |
|
| 172 |
|
| 173 |
### Framework Versions
|
| 174 |
- Python: 3.10.14
|
config_setfit.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
|
|
|
| 2 |
"labels": [
|
| 3 |
"negative",
|
| 4 |
"positive"
|
| 5 |
-
]
|
| 6 |
-
"normalize_embeddings": false
|
| 7 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"normalize_embeddings": false,
|
| 3 |
"labels": [
|
| 4 |
"negative",
|
| 5 |
"positive"
|
| 6 |
+
]
|
|
|
|
| 7 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 540795752
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cd627f226395ff80fa334ae7887ad92c07662b0c03b1aaf95b00ee4c6676390f
|
| 3 |
size 540795752
|
model_head.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 25463
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0121a33cb7938518fbc3112aab74a287a446eba1a98881177121ab7e6e14b55
|
| 3 |
size 25463
|