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
|