Sentence Similarity
Safetensors
sentence-transformers
English
PyLate
bert
ColBERT
feature-extraction
Generated from Trainer
loss:Contrastive
Eval Results (legacy)
text-embeddings-inference
Instructions to use NeuML/pubmedbert-base-colbert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use NeuML/pubmedbert-base-colbert with sentence-transformers:
from pylate import models queries = [ "Which planet is known as the Red Planet?", "What is the largest planet in our solar system?", ] documents = [ ["Mars is the Red Planet.", "Venus is Earth's twin."], ["Jupiter is the largest planet.", "Saturn has rings."], ] model = models.ColBERT(model_name_or_path="NeuML/pubmedbert-base-colbert") queries_emb = model.encode(queries, is_query=True) docs_emb = model.encode(documents, is_query=False) - Notebooks
- Google Colab
- Kaggle
File size: 254 Bytes
8f83317 | 1 2 3 4 5 6 7 8 9 10 11 | epoch,steps,accuracy
-1,-1,0.9998083710670471
-1,-1,0.9999090433120728
-1,-1,0.9998180866241455
-1,-1,0.999454140663147
-1,-1,0.9991812109947205
-1,-1,0.9993631839752197
-1,-1,0.999399721622467
-1,-1,0.9992722272872925
-1,-1,0.9998180866241455
|