IMvision12 commited on
Commit
aff6ad2
·
verified ·
1 Parent(s): e69d31f

Add Available classes section to model card

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md CHANGED
@@ -56,6 +56,23 @@ Load any ELECTRA variant the same way with `from_weights("kerasformers/<variant>
56
  | base | [`kerasformers/electra_base_discriminator`](https://huggingface.co/kerasformers/electra_base_discriminator) | [`kerasformers/electra_base_generator`](https://huggingface.co/kerasformers/electra_base_generator) |
57
  | large | [`kerasformers/electra_large_discriminator`](https://huggingface.co/kerasformers/electra_large_discriminator) | [`kerasformers/electra_large_generator`](https://huggingface.co/kerasformers/electra_large_generator) |
58
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
  ## Tips
60
 
61
  - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
 
56
  | base | [`kerasformers/electra_base_discriminator`](https://huggingface.co/kerasformers/electra_base_discriminator) | [`kerasformers/electra_base_generator`](https://huggingface.co/kerasformers/electra_base_generator) |
57
  | large | [`kerasformers/electra_large_discriminator`](https://huggingface.co/kerasformers/electra_large_discriminator) | [`kerasformers/electra_large_generator`](https://huggingface.co/kerasformers/electra_large_generator) |
58
 
59
+ ## Available classes
60
+
61
+ Load any of these from this repo with `from_weights("kerasformers/electra_small_discriminator")` (or on the fly via the `hf:` prefix). The pretrained backbone is shared; task heads not stored in this checkpoint start randomly initialized, ready for fine-tuning (or load a `hf:` fine-tune).
62
+
63
+ | Class | Task |
64
+ |---|---|
65
+ | `ElectraModel` | Encoder backbone |
66
+ | `ElectraSequenceClassify` | Sequence classification |
67
+ | `ElectraTokenClassify` | Token classification (NER / POS) |
68
+ | `ElectraQnA` | Extractive question answering |
69
+ | `ElectraMultipleChoice` | Multiple choice |
70
+
71
+ ```python
72
+ from kerasformers.models.electra import ElectraSequenceClassify
73
+ model = ElectraSequenceClassify.from_weights("kerasformers/electra_small_discriminator")
74
+ ```
75
+
76
  ## Tips
77
 
78
  - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.