Sentence Similarity
sentence-transformers
PyTorch
Transformers
Russian
bert
feature-extraction
text-embeddings-inference
Instructions to use Den4ikAI/rubert-tiny2-retriever with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use Den4ikAI/rubert-tiny2-retriever with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("Den4ikAI/rubert-tiny2-retriever") sentences = [ "query: Когда родился Пушкин?", "passage: Алекса́ндр Серге́евич Пу́шкин (26 мая [6 июня] 1799, Москва — 29 января [10 февраля] 1837, Санкт-Петербург) — русский поэт, драматург и прозаик, заложивший основы русского реалистического направления[2], литературный критик[3] и теоретик литературы, историк[3], публицист, журналист[3].", "passage: Пушкин ловил кайф со своими друзьями", "passage: Пушкин из самых авторитетных литературных деятелей первой трети XIX века. Ещё при жизни Пушкина сложилась его репутация величайшего национального русского поэта[4][5]. Пушкин рассматривается как основоположник современного русского литературного языка[~ 2]." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use Den4ikAI/rubert-tiny2-retriever with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("Den4ikAI/rubert-tiny2-retriever") model = AutoModel.from_pretrained("Den4ikAI/rubert-tiny2-retriever") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -5,7 +5,16 @@ tags:
|
|
| 5 |
- feature-extraction
|
| 6 |
- sentence-similarity
|
| 7 |
- transformers
|
| 8 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
# {MODEL_NAME}
|
|
|
|
| 5 |
- feature-extraction
|
| 6 |
- sentence-similarity
|
| 7 |
- transformers
|
| 8 |
+
license: mit
|
| 9 |
+
language:
|
| 10 |
+
- ru
|
| 11 |
+
widget:
|
| 12 |
+
- source_sentence: "query: Когда родился Пушкин?"
|
| 13 |
+
|
| 14 |
+
sentences:
|
| 15 |
+
- "passage: Алекса́ндр Серге́евич Пу́шкин (26 мая [6 июня] 1799, Москва — 29 января [10 февраля] 1837, Санкт-Петербург) — русский поэт, драматург и прозаик, заложивший основы русского реалистического направления[2], литературный критик[3] и теоретик литературы, историк[3], публицист, журналист[3]."
|
| 16 |
+
- "passage: Пушкин ловил кайф со своими друзьями"
|
| 17 |
+
- "passage: Пушкин из самых авторитетных литературных деятелей первой трети XIX века. Ещё при жизни Пушкина сложилась его репутация величайшего национального русского поэта[4][5]. Пушкин рассматривается как основоположник современного русского литературного языка[~ 2]."
|
| 18 |
---
|
| 19 |
|
| 20 |
# {MODEL_NAME}
|