Sentence Similarity
sentence-transformers
PyTorch
Safetensors
Transformers
Russian
bert
pretraining
russian
fill-mask
embeddings
masked-lm
tiny
feature-extraction
text-embeddings-inference
Instructions to use cointegrated/rubert-tiny2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use cointegrated/rubert-tiny2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("cointegrated/rubert-tiny2") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Transformers
How to use cointegrated/rubert-tiny2 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("cointegrated/rubert-tiny2") model = AutoModelForPreTraining.from_pretrained("cointegrated/rubert-tiny2") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit ·
776805c
1
Parent(s): 2ce1adc
update the readme
Browse files
README.md
CHANGED
|
@@ -9,6 +9,8 @@ tags:
|
|
| 9 |
- tiny
|
| 10 |
- feature-extraction
|
| 11 |
- sentence-similarity
|
|
|
|
|
|
|
| 12 |
|
| 13 |
license: mit
|
| 14 |
widget:
|
|
@@ -46,3 +48,12 @@ def embed_bert_cls(text, model, tokenizer):
|
|
| 46 |
print(embed_bert_cls('привет мир', model, tokenizer).shape)
|
| 47 |
# (312,)
|
| 48 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
- tiny
|
| 10 |
- feature-extraction
|
| 11 |
- sentence-similarity
|
| 12 |
+
- sentence-transformers
|
| 13 |
+
- transformers
|
| 14 |
|
| 15 |
license: mit
|
| 16 |
widget:
|
|
|
|
| 48 |
print(embed_bert_cls('привет мир', model, tokenizer).shape)
|
| 49 |
# (312,)
|
| 50 |
```
|
| 51 |
+
|
| 52 |
+
Alternatively, you can use the model with `sentence_transformers`:
|
| 53 |
+
```Python
|
| 54 |
+
from sentence_transformers import SentenceTransformer
|
| 55 |
+
model = SentenceTransformer('cointegrated/rubert-tiny2')
|
| 56 |
+
sentences = ["привет мир", "hello world", "здравствуй вселенная"]
|
| 57 |
+
embeddings = model.encode(sentences)
|
| 58 |
+
print(embeddings)
|
| 59 |
+
```
|