nljubesi commited on
Commit
3bb29b0
1 Parent(s): cbe72b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -84,13 +84,16 @@ model_args = ClassificationArgs(
84
  regression=True,
85
  )
86
  model = ClassificationModel(model_type="xlmroberta", model_name="classla/xlm-r-parlasent",use_cuda=torch.cuda.is_available(), num_labels=1,args=model_args)
87
- model.predict(["I fully disagree with this argument.", "The ministers are entering the chamber.", "Things can always be improved in the future.", "These are great news."])
 
 
 
88
  ```
89
 
90
  Output:
91
  ```python
92
  (
93
- array([0.11633301, 3.63671875, 4.203125 , 5.30859375]),
94
- array([0.11633301, 3.63671875, 4.203125 , 5.30859375])
95
  )
96
  ```
 
84
  regression=True,
85
  )
86
  model = ClassificationModel(model_type="xlmroberta", model_name="classla/xlm-r-parlasent",use_cuda=torch.cuda.is_available(), num_labels=1,args=model_args)
87
+ model.predict(["I fully disagree with this argument.",
88
+ "The ministers are entering the chamber.",
89
+ "Things can always be improved in the future.",
90
+ "These are great news."])
91
  ```
92
 
93
  Output:
94
  ```python
95
  (
96
+ array([0.11633301, 3.63671875, 4.203125, 5.30859375]),
97
+ array([0.11633301, 3.63671875, 4.203125, 5.30859375])
98
  )
99
  ```