Text Classification
Transformers
PyTorch
Safetensors
xlm-roberta
genre
text-genre
Inference Endpoints
TajaKuzman commited on
Commit
ab3f058
1 Parent(s): be4e3d7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
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