Sentence Similarity
sentence-transformers
ONNX
Safetensors
Transformers
Transformers.js
English
bert
feature-extraction
text-embeddings-inference
information-retrieval
knowledge-distillation
Instructions to use MongoDB/mdbr-leaf-ir with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use MongoDB/mdbr-leaf-ir with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("MongoDB/mdbr-leaf-ir") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use MongoDB/mdbr-leaf-ir with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("MongoDB/mdbr-leaf-ir") model = AutoModel.from_pretrained("MongoDB/mdbr-leaf-ir", device_map="auto") - Transformers.js
How to use MongoDB/mdbr-leaf-ir with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('sentence-similarity', 'MongoDB/mdbr-leaf-ir'); - Inference
- Notebooks
- Google Colab
- Kaggle
Upload README.md
Browse files
README.md
CHANGED
|
@@ -263,12 +263,6 @@ similarities = query_embeds.astype(int) @ doc_embeds.astype(int).T
|
|
| 263 |
print('After quantization:')
|
| 264 |
print(f"* Embeddings type: {query_embeds.dtype}")
|
| 265 |
print(f"* Similarities: \n{similarities}")
|
| 266 |
-
|
| 267 |
-
# After quantization:
|
| 268 |
-
# * Embeddings type: int8
|
| 269 |
-
# * Similarities:
|
| 270 |
-
# [[118022 79111]
|
| 271 |
-
# [ 72961 98333]]
|
| 272 |
```
|
| 273 |
|
| 274 |
<details>
|
|
@@ -284,7 +278,7 @@ After quantization:
|
|
| 284 |
```
|
| 285 |
</details>
|
| 286 |
|
| 287 |
-
# Evaluation
|
| 288 |
|
| 289 |
Please [see here](https://huggingface.co/MongoDB/mdbr-leaf-ir/blob/main/evaluate_models.ipynb).
|
| 290 |
|
|
|
|
| 263 |
print('After quantization:')
|
| 264 |
print(f"* Embeddings type: {query_embeds.dtype}")
|
| 265 |
print(f"* Similarities: \n{similarities}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 266 |
```
|
| 267 |
|
| 268 |
<details>
|
|
|
|
| 278 |
```
|
| 279 |
</details>
|
| 280 |
|
| 281 |
+
# Model Evaluation
|
| 282 |
|
| 283 |
Please [see here](https://huggingface.co/MongoDB/mdbr-leaf-ir/blob/main/evaluate_models.ipynb).
|
| 284 |
|