Update README.md
Browse files
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.",
|
|
|
|
|
|
|
88 |
```
|
89 |
|
90 |
Output:
|
91 |
```python
|
92 |
(
|
93 |
-
array([0.11633301, 3.63671875, 4.203125
|
94 |
-
array([0.11633301, 3.63671875, 4.203125
|
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 |
```
|