Update README.md (#3)
Browse files- Update README.md (a9140e55b1245aaf037f536fddae7a1bfd8612dc)
Co-authored-by: Daniel Tamayo <[email protected]>
README.md
CHANGED
|
@@ -79,24 +79,24 @@ Training Hyperparemeters
|
|
| 79 |
'sequence': 'Adoro menjar un bon plat de marisc al costat de la platja.',
|
| 80 |
'token': 31717,
|
| 81 |
'token_str': 'marisc'}]
|
| 82 |
-
>>> pprint(unmasker("Intento a la platja de<mask>cada any, és fantástica.",top_k=3))
|
| 83 |
-
[{'score': 0.
|
| 84 |
-
'sequence': 'Intento a la platja de Pals cada any, és fantástica.',
|
| 85 |
'token': 28365,
|
| 86 |
'token_str': 'Pals'},
|
| 87 |
-
{'score': 0.
|
| 88 |
-
'sequence': 'Intento a la platja de
|
| 89 |
-
'token': 11420,
|
| 90 |
-
'token_str': 'Lloret'},
|
| 91 |
-
{'score': 0.03042813576757908,
|
| 92 |
-
'sequence': 'Intento a la platja de Calella cada any, és fantástica.',
|
| 93 |
'token': 11472,
|
| 94 |
-
'token_str': 'Calella'}
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
```
|
| 96 |
|
| 97 |
Which is equivalent to the following torch script:
|
| 98 |
|
| 99 |
-
```
|
| 100 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 101 |
import torch
|
| 102 |
|
|
|
|
| 79 |
'sequence': 'Adoro menjar un bon plat de marisc al costat de la platja.',
|
| 80 |
'token': 31717,
|
| 81 |
'token_str': 'marisc'}]
|
| 82 |
+
>>> pprint(unmasker("Intento anar a la platja de<mask>cada any, és fantástica.",top_k=3))
|
| 83 |
+
[{'score': 0.06159511208534241,
|
| 84 |
+
'sequence': 'Intento anar a la platja de Pals cada any, és fantástica.',
|
| 85 |
'token': 28365,
|
| 86 |
'token_str': 'Pals'},
|
| 87 |
+
{'score': 0.04985760524868965,
|
| 88 |
+
'sequence': 'Intento anar a la platja de Calella cada any, és fantástica.',
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
'token': 11472,
|
| 90 |
+
'token_str': 'Calella'},
|
| 91 |
+
{'score': 0.048444587737321854,
|
| 92 |
+
'sequence': 'Intento anar a la platja de Lloret cada any, és fantástica.',
|
| 93 |
+
'token': 11420,
|
| 94 |
+
'token_str': 'Lloret'}]
|
| 95 |
```
|
| 96 |
|
| 97 |
Which is equivalent to the following torch script:
|
| 98 |
|
| 99 |
+
```python
|
| 100 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 101 |
import torch
|
| 102 |
|