Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,9 @@
|
|
1 |
---
|
|
|
2 |
license: mit
|
|
|
|
|
|
|
3 |
---
|
4 |
# Zero-shot text classification (model based on albert-xxlarge-v2) trained with self-supervised tuning
|
5 |
|
@@ -26,7 +30,8 @@ There are three versions of models released. The details are:
|
|
26 |
| Model | Backbone | #params | accuracy | Speed | #Training data
|
27 |
|------------|-----------|----------|-------|-------|----|
|
28 |
| [zero-shot-classify-SSTuning-base](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-base) | [roberta-base](https://huggingface.co/roberta-base) | 125M | Low | High | 20.48M |
|
29 |
-
| [zero-shot-classify-SSTuning-large](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-large) | [roberta-large](
|
|
|
30 |
| [zero-shot-classify-SSTuning-ALBERT](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT) | [albert-xxlarge-v2](https://huggingface.co/albert-xxlarge-v2) | 235M | High | Low| 5.12M |
|
31 |
|
32 |
Please note that zero-shot-classify-SSTuning-base is trained with more data (20.48M) than the paper, as this will increase the accuracy.
|
|
|
1 |
---
|
2 |
+
inference: false
|
3 |
license: mit
|
4 |
+
tags:
|
5 |
+
- Zero-Shot Classification
|
6 |
+
pipeline_tag: zero-shot-classification
|
7 |
---
|
8 |
# Zero-shot text classification (model based on albert-xxlarge-v2) trained with self-supervised tuning
|
9 |
|
|
|
30 |
| Model | Backbone | #params | accuracy | Speed | #Training data
|
31 |
|------------|-----------|----------|-------|-------|----|
|
32 |
| [zero-shot-classify-SSTuning-base](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-base) | [roberta-base](https://huggingface.co/roberta-base) | 125M | Low | High | 20.48M |
|
33 |
+
| [zero-shot-classify-SSTuning-large](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-large) | [roberta-large](
|
34 |
+
https://huggingface.co/roberta-large) | 355M | Medium | Medium | 5.12M |
|
35 |
| [zero-shot-classify-SSTuning-ALBERT](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT) | [albert-xxlarge-v2](https://huggingface.co/albert-xxlarge-v2) | 235M | High | Low| 5.12M |
|
36 |
|
37 |
Please note that zero-shot-classify-SSTuning-base is trained with more data (20.48M) than the paper, as this will increase the accuracy.
|