Commit
·
b8eb90a
1
Parent(s):
053a766
Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ This is best-base-cased model trained on the binary dataset prepared for adverti
|
|
8 |
|
9 |
## Example of classification
|
10 |
|
11 |
-
|
12 |
from transformers import AutoModelForSequenceClassification
|
13 |
from transformers import AutoTokenizer
|
14 |
import numpy as np
|
@@ -22,7 +22,7 @@ scores = output[0][0].detach().to('cpu').numpy()
|
|
22 |
scores = softmax(scores)
|
23 |
prediction_class = np.argmax(scores)
|
24 |
print(prediction_class)
|
25 |
-
|
26 |
Output:
|
27 |
```
|
28 |
1
|
|
|
8 |
|
9 |
## Example of classification
|
10 |
|
11 |
+
```python
|
12 |
from transformers import AutoModelForSequenceClassification
|
13 |
from transformers import AutoTokenizer
|
14 |
import numpy as np
|
|
|
22 |
scores = softmax(scores)
|
23 |
prediction_class = np.argmax(scores)
|
24 |
print(prediction_class)
|
25 |
+
```
|
26 |
Output:
|
27 |
```
|
28 |
1
|