Commit
•
2612cbd
1
Parent(s):
64364f0
Add SetFit model
Browse files- .gitattributes +1 -0
- 1_Pooling/config.json +10 -0
- 2_Dense/config.json +1 -0
- 2_Dense/pytorch_model.bin +3 -0
- README.md +239 -1
- config.json +32 -0
- config_sentence_transformers.json +9 -0
- config_setfit.json +9 -0
- model.safetensors +3 -0
- model_head.pkl +3 -0
- modules.json +26 -0
- sentence_bert_config.json +4 -0
- special_tokens_map.json +37 -0
- tokenizer.json +3 -0
- tokenizer_config.json +58 -0
- vocab.txt +0 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
1_Pooling/config.json
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"word_embedding_dimension": 768,
|
3 |
+
"pooling_mode_cls_token": true,
|
4 |
+
"pooling_mode_mean_tokens": false,
|
5 |
+
"pooling_mode_max_tokens": false,
|
6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
7 |
+
"pooling_mode_weightedmean_tokens": false,
|
8 |
+
"pooling_mode_lasttoken": false,
|
9 |
+
"include_prompt": true
|
10 |
+
}
|
2_Dense/config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"in_features": 768, "out_features": 768, "bias": true, "activation_function": "torch.nn.modules.activation.Tanh"}
|
2_Dense/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4b4be681fccad796fda2017043f21a0ed4b080c3c8ce41330bf57db76c028d53
|
3 |
+
size 2364028
|
README.md
CHANGED
@@ -1,3 +1,241 @@
|
|
1 |
---
|
2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
library_name: setfit
|
3 |
+
tags:
|
4 |
+
- setfit
|
5 |
+
- sentence-transformers
|
6 |
+
- text-classification
|
7 |
+
- generated_from_setfit_trainer
|
8 |
+
metrics:
|
9 |
+
- accuracy
|
10 |
+
widget:
|
11 |
+
- text: حصل لاعبو المنتخب السوري لكرة القدم على مكافأة سريعة، بعد تحقيقهم الفوز على
|
12 |
+
نظيرهم السعودي 2/1 في المباراة التي أقيمت، أول من أمس، على استاد الريان ضمن مباريات
|
13 |
+
المجموعة الثانية لبطولة أمم آسيا بكرة القدم، المقامة حالياً في الدوحة وتستمر حتى
|
14 |
+
27 يناير الجاري.
|
15 |
+
- text: هدد الامين العام لحزب الله حسن نصرالله الثلاثاء بضرب البنى التحتية في اسرائيل
|
16 |
+
وبتدمير ابنية في تل ابيب في حال هاجمت الدولة العبرية لبنان، ملمحا الى اقتناء حزب
|
17 |
+
الله القدرات العسكرية التي تخوله القيام بذلك.
|
18 |
+
- text: برّأت لجنة تحقيق إسرائيلية على نطاق واسع أمس الجيش الإسرائيلي في تحقيق بشأن
|
19 |
+
حصار غزة والعدوان على اسطول الحرية، فيما كشف تقرير إخباري إسرائيلي أن وزير الخارجية
|
20 |
+
أفيغدور ليبرمان، رسم خريطة لدولة فلسطينية بحدود مؤقتة، سارع الفلسطينيون برفضها.
|
21 |
+
- text: دعا الرئيس المنتخب لساحل العاج الحسن وتارا، أمس، الى استخدام القوة لحمل منافسه
|
22 |
+
لوران غباغبو على التنحي.
|
23 |
+
- text: تتجه الأنظار، اليوم، إلى القمة الكلاسيكية الخاصة بين منتخبي اليابان وكوريا
|
24 |
+
الجنوبية، على ملعب نادي الغرافة في نصف نهائي كأس آسيا الخامسة عشرة لكرة القدم
|
25 |
+
في الدوحة.
|
26 |
+
pipeline_tag: text-classification
|
27 |
+
inference: true
|
28 |
+
base_model: sentence-transformers/LaBSE
|
29 |
+
model-index:
|
30 |
+
- name: SetFit with sentence-transformers/LaBSE
|
31 |
+
results:
|
32 |
+
- task:
|
33 |
+
type: text-classification
|
34 |
+
name: Text Classification
|
35 |
+
dataset:
|
36 |
+
name: Unknown
|
37 |
+
type: unknown
|
38 |
+
split: test
|
39 |
+
metrics:
|
40 |
+
- type: accuracy
|
41 |
+
value: 0.9083333333333333
|
42 |
+
name: Accuracy
|
43 |
---
|
44 |
+
|
45 |
+
# SetFit with sentence-transformers/LaBSE
|
46 |
+
|
47 |
+
This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [sentence-transformers/LaBSE](https://huggingface.co/sentence-transformers/LaBSE) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
|
48 |
+
|
49 |
+
The model has been trained using an efficient few-shot learning technique that involves:
|
50 |
+
|
51 |
+
1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
|
52 |
+
2. Training a classification head with features from the fine-tuned Sentence Transformer.
|
53 |
+
|
54 |
+
## Model Details
|
55 |
+
|
56 |
+
### Model Description
|
57 |
+
- **Model Type:** SetFit
|
58 |
+
- **Sentence Transformer body:** [sentence-transformers/LaBSE](https://huggingface.co/sentence-transformers/LaBSE)
|
59 |
+
- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
|
60 |
+
- **Maximum Sequence Length:** 256 tokens
|
61 |
+
- **Number of Classes:** 4 classes
|
62 |
+
<!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
|
63 |
+
<!-- - **Language:** Unknown -->
|
64 |
+
<!-- - **License:** Unknown -->
|
65 |
+
|
66 |
+
### Model Sources
|
67 |
+
|
68 |
+
- **Repository:** [SetFit on GitHub](https://github.com/huggingface/setfit)
|
69 |
+
- **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
|
70 |
+
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
71 |
+
|
72 |
+
### Model Labels
|
73 |
+
| Label | Examples |
|
74 |
+
|:---------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
75 |
+
| life | <ul><li>'عارضات يرتدين ابتكارات مختلفة لعدد من المصممين خلال أسبوع ساو باولو للموضة في البرازيل.'</li><li>'رفض قاض أميركي أمس، دعوى التشهير التي أقامها ديفيد بيكهام مطالبا بتعويض قدره 25 مليون دولار ضد مجلة تتناول أخبار المشاهير نشرت مقالا يقول أن نجم كرة القدم الإنجليزي المتزوج مارس الجنس مع'</li><li>'تعرض صالة «هالسيون غاليري» في لندن اعتباراً من السبت، لوحات للموسيقي الاميركي بوب ديلان في تكملة لمجموعته الفنية «درون بلانك سيريز» التي سبق عرضها في ألمانيا في 2007 وبريطانيا في 2008.'</li></ul> |
|
76 |
+
| politics | <ul><li>'لقي شخص واحد على الأقل مصرعه، واعتبر ثلاثة آخرون في عداد المفقودين، بعد انفجار طائرة ركاب روسية، بعد قليل من هبوطها اضطرارياً في مطار مدينة "سورغوت" بمنطقة سيبيريا، وعلى متنها نحو 130 راكباً، وفق ما أكدت مصادر رسمية امس.'</li><li>'اعلنت الشرطة الباكستانية ان خمسة اشخاص على الاقل قتلوا في تفجير قنبلة خبئت في دراجة نارية في منطقة تقع شمال غرب باكستان تشهد باستمرار اعمال عنف بين السنة والشيعة.'</li><li>'وحده الأحمق يسعى إلى التنبؤ بمسار الانتفاضتين التونسية والمصرية، لكن الغبي فقط ينكر ان ثمة شيئاً دراماتيكياً وعميقاً يجري على نطاق واسع في العالمين العربي والإسلامي، وهو أكثر من مجرد ظاهرة سياسية.'</li></ul> |
|
77 |
+
| sports | <ul><li>'زار سمو الشيخ عبدالله بن زايد ال نهيان وزير الخارجية بعثة المنتخب الوطني الاول لكرة القدم المتواجدة في العاصمة القطرية الدوحة للمشاركة في كاس اسيا 2011.'</li><li>'أعلنت أكاديمية لوريوس العالمية أسماء الفائزين بجوائزها السنوية الثانية عشرة لعام ،2011 في احتفال ضخم أقيم مساء أمس بفندق قصر الإمارات، بالعاصمة أبوظبي، بحضور نخبة كبيرة من نجوم ومشاهير العالم في الرياضة والفن.'</li><li>'حقق فريق الشارقة فوزا هاما ومستحقا على النصر بنتيجة 4/ 1، في المباراة التي جمعتهما مساء امس على ستاد ال مكتوم بدبي ضمن الجولة 15 من دوري اتصالات.'</li></ul> |
|
78 |
+
| business | <ul><li>'بلغ مجمل عدد الشركات التي تستثمر في قطاع الخدمات الأمنية في مختلف أنحاء الدولة 404 شركات، توظف لديها نحو 30 ألف موظف، بحسب ورقة عمل استعرضها مدير إدارة شركات الأمن الخاصة في وزارة الداخلية، العقيد أحمد الحنطوبي، خلال ملتقى «الأمن وحماية الاستثمارات»، الذي عقد في غرفة تجارة وصناعة الشارقة، أمس.'</li><li>'أعلنت شركة «طيران الإمارات» توسيع شبكة خطوطها في شمال أوروبا، من خلال إضافة مزيد من الرحلات المنتظمة إلى خدماتها القائمة إلى كل من النمسا وألمانيا في مارس وسبتمبر المقبلين، إضافة إلى الخدمات الجديدة التي أعلنتها أخيراً إلى كل من سويسرا والدنمارك.'</li><li>'وقّعت شركة أبوظبي للخدمات العامة «مساندة»، أمس، اتفاقية شراكة استراتيجية من الفئة الذهبية مع شركة «أوراكل» العالمية المتخصصة في البرامج التقنية، ما سيدعم إدارة خدمات تكنولوجيا المعلومات في «مساندة» في تنفيذ المشروعات المنوطة إليها، بما فيها برنامج تخطيط موارد المشروعات الحكومية المشتركة.'</li></ul> |
|
79 |
+
|
80 |
+
## Evaluation
|
81 |
+
|
82 |
+
### Metrics
|
83 |
+
| Label | Accuracy |
|
84 |
+
|:--------|:---------|
|
85 |
+
| **all** | 0.9083 |
|
86 |
+
|
87 |
+
## Uses
|
88 |
+
|
89 |
+
### Direct Use for Inference
|
90 |
+
|
91 |
+
First install the SetFit library:
|
92 |
+
|
93 |
+
```bash
|
94 |
+
pip install setfit
|
95 |
+
```
|
96 |
+
|
97 |
+
Then you can load this model and run inference.
|
98 |
+
|
99 |
+
```python
|
100 |
+
from setfit import SetFitModel
|
101 |
+
|
102 |
+
# Download from the 🤗 Hub
|
103 |
+
model = SetFitModel.from_pretrained("Ezzaldin-97/LaBSE-based-Arabic-News-Classifier")
|
104 |
+
# Run inference
|
105 |
+
preds = model("دعا الرئيس المنتخب لساحل العاج الحسن وتارا، أمس، الى استخدام القوة لحمل منافسه لوران غباغبو على التنحي.")
|
106 |
+
```
|
107 |
+
|
108 |
+
<!--
|
109 |
+
### Downstream Use
|
110 |
+
|
111 |
+
*List how someone could finetune this model on their own dataset.*
|
112 |
+
-->
|
113 |
+
|
114 |
+
<!--
|
115 |
+
### Out-of-Scope Use
|
116 |
+
|
117 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
118 |
+
-->
|
119 |
+
|
120 |
+
<!--
|
121 |
+
## Bias, Risks and Limitations
|
122 |
+
|
123 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
124 |
+
-->
|
125 |
+
|
126 |
+
<!--
|
127 |
+
### Recommendations
|
128 |
+
|
129 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
130 |
+
-->
|
131 |
+
|
132 |
+
## Training Details
|
133 |
+
|
134 |
+
### Training Set Metrics
|
135 |
+
| Training set | Min | Median | Max |
|
136 |
+
|:-------------|:----|:--------|:----|
|
137 |
+
| Word count | 11 | 32.9082 | 60 |
|
138 |
+
|
139 |
+
| Label | Training Sample Count |
|
140 |
+
|:---------|:----------------------|
|
141 |
+
| business | 54 |
|
142 |
+
| life | 98 |
|
143 |
+
| politics | 91 |
|
144 |
+
| sports | 73 |
|
145 |
+
|
146 |
+
### Training Hyperparameters
|
147 |
+
- batch_size: (16, 16)
|
148 |
+
- num_epochs: (2, 2)
|
149 |
+
- max_steps: -1
|
150 |
+
- sampling_strategy: oversampling
|
151 |
+
- num_iterations: 20
|
152 |
+
- body_learning_rate: (2e-05, 2e-05)
|
153 |
+
- head_learning_rate: 2e-05
|
154 |
+
- loss: CosineSimilarityLoss
|
155 |
+
- distance_metric: cosine_distance
|
156 |
+
- margin: 0.25
|
157 |
+
- end_to_end: False
|
158 |
+
- use_amp: False
|
159 |
+
- warmup_proportion: 0.1
|
160 |
+
- seed: 42
|
161 |
+
- eval_max_steps: -1
|
162 |
+
- load_best_model_at_end: False
|
163 |
+
|
164 |
+
### Training Results
|
165 |
+
| Epoch | Step | Training Loss | Validation Loss |
|
166 |
+
|:------:|:----:|:-------------:|:---------------:|
|
167 |
+
| 0.0013 | 1 | 0.2934 | - |
|
168 |
+
| 0.0633 | 50 | 0.2025 | - |
|
169 |
+
| 0.1266 | 100 | 0.104 | - |
|
170 |
+
| 0.1899 | 150 | 0.0906 | - |
|
171 |
+
| 0.2532 | 200 | 0.0012 | - |
|
172 |
+
| 0.3165 | 250 | 0.001 | - |
|
173 |
+
| 0.3797 | 300 | 0.0008 | - |
|
174 |
+
| 0.4430 | 350 | 0.0007 | - |
|
175 |
+
| 0.5063 | 400 | 0.0005 | - |
|
176 |
+
| 0.5696 | 450 | 0.0006 | - |
|
177 |
+
| 0.6329 | 500 | 0.0003 | - |
|
178 |
+
| 0.6962 | 550 | 0.0003 | - |
|
179 |
+
| 0.7595 | 600 | 0.0004 | - |
|
180 |
+
| 0.8228 | 650 | 0.0004 | - |
|
181 |
+
| 0.8861 | 700 | 0.0003 | - |
|
182 |
+
| 0.9494 | 750 | 0.0003 | - |
|
183 |
+
| 1.0127 | 800 | 0.0003 | - |
|
184 |
+
| 1.0759 | 850 | 0.0002 | - |
|
185 |
+
| 1.1392 | 900 | 0.0002 | - |
|
186 |
+
| 1.2025 | 950 | 0.0003 | - |
|
187 |
+
| 1.2658 | 1000 | 0.0003 | - |
|
188 |
+
| 1.3291 | 1050 | 0.0002 | - |
|
189 |
+
| 1.3924 | 1100 | 0.0002 | - |
|
190 |
+
| 1.4557 | 1150 | 0.0004 | - |
|
191 |
+
| 1.5190 | 1200 | 0.0002 | - |
|
192 |
+
| 1.5823 | 1250 | 0.0003 | - |
|
193 |
+
| 1.6456 | 1300 | 0.0002 | - |
|
194 |
+
| 1.7089 | 1350 | 0.0002 | - |
|
195 |
+
| 1.7722 | 1400 | 0.0003 | - |
|
196 |
+
| 1.8354 | 1450 | 0.0002 | - |
|
197 |
+
| 1.8987 | 1500 | 0.0002 | - |
|
198 |
+
| 1.9620 | 1550 | 0.0003 | - |
|
199 |
+
|
200 |
+
### Framework Versions
|
201 |
+
- Python: 3.10.12
|
202 |
+
- SetFit: 1.0.3
|
203 |
+
- Sentence Transformers: 2.6.1
|
204 |
+
- Transformers: 4.38.2
|
205 |
+
- PyTorch: 2.2.1+cu121
|
206 |
+
- Datasets: 2.18.0
|
207 |
+
- Tokenizers: 0.15.2
|
208 |
+
|
209 |
+
## Citation
|
210 |
+
|
211 |
+
### BibTeX
|
212 |
+
```bibtex
|
213 |
+
@article{https://doi.org/10.48550/arxiv.2209.11055,
|
214 |
+
doi = {10.48550/ARXIV.2209.11055},
|
215 |
+
url = {https://arxiv.org/abs/2209.11055},
|
216 |
+
author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
|
217 |
+
keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
|
218 |
+
title = {Efficient Few-Shot Learning Without Prompts},
|
219 |
+
publisher = {arXiv},
|
220 |
+
year = {2022},
|
221 |
+
copyright = {Creative Commons Attribution 4.0 International}
|
222 |
+
}
|
223 |
+
```
|
224 |
+
|
225 |
+
<!--
|
226 |
+
## Glossary
|
227 |
+
|
228 |
+
*Clearly define terms in order to be accessible across audiences.*
|
229 |
+
-->
|
230 |
+
|
231 |
+
<!--
|
232 |
+
## Model Card Authors
|
233 |
+
|
234 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
235 |
+
-->
|
236 |
+
|
237 |
+
<!--
|
238 |
+
## Model Card Contact
|
239 |
+
|
240 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
241 |
+
-->
|
config.json
ADDED
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "sentence-transformers/LaBSE",
|
3 |
+
"architectures": [
|
4 |
+
"BertModel"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.1,
|
7 |
+
"classifier_dropout": null,
|
8 |
+
"directionality": "bidi",
|
9 |
+
"gradient_checkpointing": false,
|
10 |
+
"hidden_act": "gelu",
|
11 |
+
"hidden_dropout_prob": 0.1,
|
12 |
+
"hidden_size": 768,
|
13 |
+
"initializer_range": 0.02,
|
14 |
+
"intermediate_size": 3072,
|
15 |
+
"layer_norm_eps": 1e-12,
|
16 |
+
"max_position_embeddings": 512,
|
17 |
+
"model_type": "bert",
|
18 |
+
"num_attention_heads": 12,
|
19 |
+
"num_hidden_layers": 12,
|
20 |
+
"pad_token_id": 0,
|
21 |
+
"pooler_fc_size": 768,
|
22 |
+
"pooler_num_attention_heads": 12,
|
23 |
+
"pooler_num_fc_layers": 3,
|
24 |
+
"pooler_size_per_head": 128,
|
25 |
+
"pooler_type": "first_token_transform",
|
26 |
+
"position_embedding_type": "absolute",
|
27 |
+
"torch_dtype": "float32",
|
28 |
+
"transformers_version": "4.38.2",
|
29 |
+
"type_vocab_size": 2,
|
30 |
+
"use_cache": true,
|
31 |
+
"vocab_size": 501153
|
32 |
+
}
|
config_sentence_transformers.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"__version__": {
|
3 |
+
"sentence_transformers": "2.0.0",
|
4 |
+
"transformers": "4.7.0",
|
5 |
+
"pytorch": "1.9.0+cu102"
|
6 |
+
},
|
7 |
+
"prompts": {},
|
8 |
+
"default_prompt_name": null
|
9 |
+
}
|
config_setfit.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"normalize_embeddings": false,
|
3 |
+
"labels": [
|
4 |
+
"business",
|
5 |
+
"life",
|
6 |
+
"politics",
|
7 |
+
"sports"
|
8 |
+
]
|
9 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6f262b5e5c2b6fedf00566169de1f2891637c4bc67f26c44d23caf3d18810afd
|
3 |
+
size 1883730160
|
model_head.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a442fa6cc8a7f856ed464efa025b5d563663758cb91e9f9950fa3d27b093eda7
|
3 |
+
size 25559
|
modules.json
ADDED
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[
|
2 |
+
{
|
3 |
+
"idx": 0,
|
4 |
+
"name": "0",
|
5 |
+
"path": "",
|
6 |
+
"type": "sentence_transformers.models.Transformer"
|
7 |
+
},
|
8 |
+
{
|
9 |
+
"idx": 1,
|
10 |
+
"name": "1",
|
11 |
+
"path": "1_Pooling",
|
12 |
+
"type": "sentence_transformers.models.Pooling"
|
13 |
+
},
|
14 |
+
{
|
15 |
+
"idx": 2,
|
16 |
+
"name": "2",
|
17 |
+
"path": "2_Dense",
|
18 |
+
"type": "sentence_transformers.models.Dense"
|
19 |
+
},
|
20 |
+
{
|
21 |
+
"idx": 3,
|
22 |
+
"name": "3",
|
23 |
+
"path": "3_Normalize",
|
24 |
+
"type": "sentence_transformers.models.Normalize"
|
25 |
+
}
|
26 |
+
]
|
sentence_bert_config.json
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"max_seq_length": 256,
|
3 |
+
"do_lower_case": false
|
4 |
+
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cls_token": {
|
3 |
+
"content": "[CLS]",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"mask_token": {
|
10 |
+
"content": "[MASK]",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": {
|
17 |
+
"content": "[PAD]",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"sep_token": {
|
24 |
+
"content": "[SEP]",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
},
|
30 |
+
"unk_token": {
|
31 |
+
"content": "[UNK]",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false
|
36 |
+
}
|
37 |
+
}
|
tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:92262b29204f8fdc169a63f9005a0e311a16262cef4d96ecfe2a7ed638662ed3
|
3 |
+
size 13632172
|
tokenizer_config.json
ADDED
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"added_tokens_decoder": {
|
3 |
+
"0": {
|
4 |
+
"content": "[PAD]",
|
5 |
+
"lstrip": false,
|
6 |
+
"normalized": false,
|
7 |
+
"rstrip": false,
|
8 |
+
"single_word": false,
|
9 |
+
"special": true
|
10 |
+
},
|
11 |
+
"100": {
|
12 |
+
"content": "[UNK]",
|
13 |
+
"lstrip": false,
|
14 |
+
"normalized": false,
|
15 |
+
"rstrip": false,
|
16 |
+
"single_word": false,
|
17 |
+
"special": true
|
18 |
+
},
|
19 |
+
"101": {
|
20 |
+
"content": "[CLS]",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false,
|
25 |
+
"special": true
|
26 |
+
},
|
27 |
+
"102": {
|
28 |
+
"content": "[SEP]",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false,
|
33 |
+
"special": true
|
34 |
+
},
|
35 |
+
"103": {
|
36 |
+
"content": "[MASK]",
|
37 |
+
"lstrip": false,
|
38 |
+
"normalized": false,
|
39 |
+
"rstrip": false,
|
40 |
+
"single_word": false,
|
41 |
+
"special": true
|
42 |
+
}
|
43 |
+
},
|
44 |
+
"clean_up_tokenization_spaces": true,
|
45 |
+
"cls_token": "[CLS]",
|
46 |
+
"do_basic_tokenize": true,
|
47 |
+
"do_lower_case": false,
|
48 |
+
"full_tokenizer_file": null,
|
49 |
+
"mask_token": "[MASK]",
|
50 |
+
"model_max_length": 512,
|
51 |
+
"never_split": null,
|
52 |
+
"pad_token": "[PAD]",
|
53 |
+
"sep_token": "[SEP]",
|
54 |
+
"strip_accents": null,
|
55 |
+
"tokenize_chinese_chars": true,
|
56 |
+
"tokenizer_class": "BertTokenizer",
|
57 |
+
"unk_token": "[UNK]"
|
58 |
+
}
|
vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|