SAVRN
Search Contact SAVRN

Organization · Verified on Hugging Face

NeuML

NeuML

Applying machine learning to solve everyday problems

Models in Library2
Datasets in Library0
Models on Hugging Face84
Followers173

Models

Model · Sentence similarity

pubmedbert-base-embeddings

NeuML

This is a PubMedBERT-base model fined-tuned using sentence-transformers. It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search. The training dataset was generated using a random sample of PubMed title-abstract pairs along with similar title pairs. PubMedBERT Embeddings produces higher quality embeddings than generalized models for medical literature. Further fine-tuning for a medical subdomain will result in even better performance. This model can be used to build embeddings databases with txtai for semantic search and/or as a knowledge source for retrieval augmented generation (RAG). Alternatively, the model can…

Open weights apache-2.0 109M parameters 512 tokens sentence-transformers

Model · Question answering

bert-small-cord19qa

NeuML

bert-small-cord19-squad model fine-tuned on the CORD-19 QA dataset. The CORD-19 QA dataset is a SQuAD 2.0 formatted list of question, context, answer combinations covering the CORD-19 dataset.

Open weights 512 tokens transformers