5roop commited on
Commit
3c3e1a4
1 Parent(s): 3e57b17

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -61,4 +61,23 @@ Results reported were obtained from 10 fine-tuning runs.
61
  test dataset | R^2
62
  --- | ---
63
  BCS | 0.6146 ± 0.0104
64
- EN | 0.6722 ± 0.0100
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
  test dataset | R^2
62
  --- | ---
63
  BCS | 0.6146 ± 0.0104
64
+ EN | 0.6722 ± 0.0100
65
+
66
+ ## Example
67
+
68
+ With `simpletransformers==0.64.3`.
69
+ ```python
70
+ from simpletransformers.classification import ClassificationModel, ClassificationArgs
71
+ import torch
72
+ model_args = ClassificationArgs(
73
+ regression=True,
74
+ )
75
+ model = ClassificationModel(model_type="xlmroberta", model_name="5roop/xlm-r-parlasent",use_cuda=torch.cuda.is_available(), num_labels=1,args=model_args)
76
+ model.predict(["""Poštovani potpredsjedničke Vlade i ministre hrvatskih branitelja, mislite li
77
+ da ste zapravo iznevjerili svoje suborce s kojima ste 555 dana prosvjedovali
78
+ u šatoru protiv tadašnjih dužnosnika jer ste zapravo donijeli zakon koji je
79
+ neprovediv, a birali ste si suradnike koji nemaju etički integritet."""])
80
+ ```
81
+
82
+ Output:
83
+ ``` (array(-0.0847168), array(-0.0847168))```