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
|
@@ -69,7 +69,7 @@ print(model.similarity(embeddings, embeddings))
|
|
| 69 |
|
| 70 |
| Model Name | AILA Statutes | Argu Ana | Legal Bench Corporate Lobbying | SCIDOCS | Stack Overflow QA | Statcan Dialogue Dataset Retrieval | Wikipedia Retrieval Multilingual | Average |
|
| 71 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 72 |
-
| bge-m3 | 0,
|
| 73 |
| **rubert-tiny-retriever** | 0,161 | 0,432 | 0,862 | 0,094 | 0,454 | 0,103 | 0,880 | 0,426 |
|
| 74 |
| rubert-tiny-turbo | 0,136 | 0,320 | 0,700 | 0,041 | 0,320 | 0,007 | 0,298 | 0,260 |
|
| 75 |
| rubert-tiny2 | 0,138 | 0,277 | 0,602 | 0,012 | 0,200 | 0,004 | 0,145 | 0,197 |
|
|
|
|
| 69 |
|
| 70 |
| Model Name | AILA Statutes | Argu Ana | Legal Bench Corporate Lobbying | SCIDOCS | Stack Overflow QA | Statcan Dialogue Dataset Retrieval | Wikipedia Retrieval Multilingual | Average |
|
| 71 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 72 |
+
| bge-m3 | 0,298 | 0,539 | 0,904 | 0,164 | 0,806 | 0,284 | 0,924 | 0,560 |
|
| 73 |
| **rubert-tiny-retriever** | 0,161 | 0,432 | 0,862 | 0,094 | 0,454 | 0,103 | 0,880 | 0,426 |
|
| 74 |
| rubert-tiny-turbo | 0,136 | 0,320 | 0,700 | 0,041 | 0,320 | 0,007 | 0,298 | 0,260 |
|
| 75 |
| rubert-tiny2 | 0,138 | 0,277 | 0,602 | 0,012 | 0,200 | 0,004 | 0,145 | 0,197 |
|