TajaKuzman
commited on
Commit
•
ab3f058
1
Parent(s):
be4e3d7
Update README.md
Browse files
README.md
CHANGED
@@ -176,6 +176,7 @@ model_args= {
|
|
176 |
"num_train_epochs": 15,
|
177 |
"learning_rate": 1e-5,
|
178 |
"max_seq_length": 512,
|
|
|
179 |
}
|
180 |
model = ClassificationModel(
|
181 |
"xlmroberta", "TajaKuzman/xlm-roberta-base-multilingual-text-genres", use_cuda=True,
|
@@ -193,6 +194,8 @@ predictions
|
|
193 |
|
194 |
```
|
195 |
|
|
|
|
|
196 |
## Performance
|
197 |
|
198 |
### Comparison with other models at in-dataset and cross-dataset experiments
|
|
|
176 |
"num_train_epochs": 15,
|
177 |
"learning_rate": 1e-5,
|
178 |
"max_seq_length": 512,
|
179 |
+
"silent": True
|
180 |
}
|
181 |
model = ClassificationModel(
|
182 |
"xlmroberta", "TajaKuzman/xlm-roberta-base-multilingual-text-genres", use_cuda=True,
|
|
|
194 |
|
195 |
```
|
196 |
|
197 |
+
Use example for prediction on a dataset, using batch processing, is available via [Google Collab](https://colab.research.google.com/drive/1yC4L_p2t3oMViC37GqSjJynQH-EWyhLr?usp=sharing).
|
198 |
+
|
199 |
## Performance
|
200 |
|
201 |
### Comparison with other models at in-dataset and cross-dataset experiments
|