Update README.md
Browse files
README.md
CHANGED
|
@@ -44,8 +44,11 @@ fill_mask("Il [MASK] ha chiesto revocarsi l'obbligo di pagamento")
|
|
| 44 |
```
|
| 45 |
here how to use it for sentence similarity
|
| 46 |
```python
|
|
|
|
|
|
|
|
|
|
| 47 |
import seaborn as sns
|
| 48 |
-
import matplotlib.pyplot as
|
| 49 |
from textwrap import wrap
|
| 50 |
|
| 51 |
#Mean Pooling - Take attention mask into account for correct averaging
|
|
|
|
| 44 |
```
|
| 45 |
here how to use it for sentence similarity
|
| 46 |
```python
|
| 47 |
+
from transformers import AutoTokenizer, AutoModel
|
| 48 |
+
from sklearn.metrics.pairwise import cosine_similarity
|
| 49 |
+
import torch
|
| 50 |
import seaborn as sns
|
| 51 |
+
import matplotlib.pyplot as plt
|
| 52 |
from textwrap import wrap
|
| 53 |
|
| 54 |
#Mean Pooling - Take attention mask into account for correct averaging
|