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: 3,790 Bytes
8f83317 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | epoch,steps,accuracy
0.13025921583952066,2000,0.9988450407981873
0.2605184316790413,4000,0.999037504196167
0.3907776475185619,6000,0.9988450407981873
0.5210368633580826,8000,0.9994224905967712
0.6512960791976032,10000,0.9996150135993958
0.7815552950371238,12000,0.9992300271987915
0.9118145108766446,14000,0.9992300271987915
0.09512937595129375,2000,0.9996359348297119
0.1902587519025875,4000,1.0
0.2853881278538813,6000,1.0
0.380517503805175,8000,1.0
0.4756468797564688,10000,0.999817967414856
0.5707762557077626,12000,0.999817967414856
0.6659056316590564,14000,0.999817967414856
0.76103500761035,16000,0.999817967414856
0.8561643835616438,18000,0.999817967414856
0.9512937595129376,20000,0.999817967414856
0.09512937595129375,2000,0.999089777469635
0.1902587519025875,4000,0.9994539022445679
0.2853881278538813,6000,0.9994539022445679
0.380517503805175,8000,0.9994539022445679
0.4756468797564688,10000,0.9994539022445679
0.5707762557077626,12000,0.9994539022445679
0.6659056316590564,14000,0.9994539022445679
0.76103500761035,16000,0.9996359348297119
0.8561643835616438,18000,0.9996359348297119
0.9512937595129376,20000,0.9996359348297119
0.09512937595129375,2000,0.9983615875244141
0.1902587519025875,4000,0.9987256526947021
0.2853881278538813,6000,0.9987256526947021
0.380517503805175,8000,0.9996359348297119
0.4756468797564688,10000,0.9996359348297119
0.5707762557077626,12000,0.999817967414856
0.6659056316590564,14000,0.9996359348297119
0.76103500761035,16000,1.0
0.8561643835616438,18000,0.999817967414856
0.9512937595129376,20000,0.999817967414856
0.09512937595129375,2000,0.99817955493927
0.1902587519025875,4000,0.998907744884491
0.2853881278538813,6000,0.998907744884491
0.380517503805175,8000,0.9985436201095581
0.4756468797564688,10000,0.9983615875244141
0.5707762557077626,12000,0.9983615875244141
0.6659056316590564,14000,0.99817955493927
0.76103500761035,16000,0.9985436201095581
0.8561643835616438,18000,0.9983615875244141
0.9512937595129376,20000,0.9983615875244141
0.09512937595129375,2000,0.998907744884491
0.1902587519025875,4000,0.999271810054779
0.2853881278538813,6000,0.999089777469635
0.380517503805175,8000,0.999089777469635
0.4756468797564688,10000,0.9994539022445679
0.5707762557077626,12000,0.999271810054779
0.6659056316590564,14000,0.999271810054779
0.76103500761035,16000,0.999271810054779
0.8561643835616438,18000,0.999089777469635
0.9512937595129376,20000,0.999089777469635
0.09905403397553365,2000,0.998598575592041
0.1981080679510673,4000,0.9989989995956421
0.297162101926601,6000,0.9991992115974426
0.3962161359021346,8000,0.9989989995956421
0.49527016987766825,10000,0.9991992115974426
0.594324203853202,12000,0.9989989995956421
0.6933782378287355,14000,0.9991992115974426
0.7924322718042692,16000,0.9989989995956421
0.8914863057798029,18000,0.9989989995956421
0.9905403397553365,20000,0.9989989995956421
0.09512937595129375,2000,0.9983615875244141
0.1902587519025875,4000,0.9985436201095581
0.2853881278538813,6000,0.9987256526947021
0.380517503805175,8000,0.9987256526947021
0.4756468797564688,10000,0.999089777469635
0.5707762557077626,12000,0.999271810054779
0.6659056316590564,14000,0.9994539022445679
0.76103500761035,16000,0.998907744884491
0.8561643835616438,18000,0.9994539022445679
0.9512937595129376,20000,0.999271810054779
0.09512937595129375,2000,0.999089777469635
0.1902587519025875,4000,0.999089777469635
0.2853881278538813,6000,0.999271810054779
0.380517503805175,8000,0.999271810054779
0.4756468797564688,10000,0.9994539022445679
0.5707762557077626,12000,0.999817967414856
0.6659056316590564,14000,0.9996359348297119
0.76103500761035,16000,0.999817967414856
0.8561643835616438,18000,0.999817967414856
0.9512937595129376,20000,0.9996359348297119
|