Sentence Similarity
sentence-transformers
Safetensors
Transformers
Russian
English
bert
feature-extraction
russian
pretraining
embeddings
tiny
retrieval
mteb
text-embeddings-inference
Instructions to use sergeyzh/rubert-tiny-retriever with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use sergeyzh/rubert-tiny-retriever with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sergeyzh/rubert-tiny-retriever") sentences = [ "Это счастливый человек", "Это счастливая собака", "Это очень счастливый человек", "Сегодня солнечный день" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use sergeyzh/rubert-tiny-retriever with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sergeyzh/rubert-tiny-retriever") model = AutoModel.from_pretrained("sergeyzh/rubert-tiny-retriever") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ print(model.similarity(embeddings, embeddings))
|
|
| 58 |
Оценки модели на задачах текстового поиска для русского языка:
|
| 59 |
|
| 60 |
|
| 61 |
-
| Model Name |
|
| 62 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 63 |
| bge-m3 | 0,654 | 0,702 | 0,830 | 0,740 | 0,712 | 0,728 |
|
| 64 |
| **rubert-tiny-retriever** | 0,574 | 0,530 | 0,611 | 0,668 | 0,589 | 0,594 |
|
|
|
|
| 58 |
Оценки модели на задачах текстового поиска для русского языка:
|
| 59 |
|
| 60 |
|
| 61 |
+
| Model Name | MIRACL Reranking | MIRACL Retrival | RiaNews Retrieval | RuBQ Reranking | RuBQ Retrieval | Average |
|
| 62 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 63 |
| bge-m3 | 0,654 | 0,702 | 0,830 | 0,740 | 0,712 | 0,728 |
|
| 64 |
| **rubert-tiny-retriever** | 0,574 | 0,530 | 0,611 | 0,668 | 0,589 | 0,594 |
|