Sentence Similarity
sentence-transformers
PyTorch
Safetensors
Transformers
bert
feature-extraction
text-embeddings-inference
Instructions to use nuvocare/WikiMedical_sent_biobert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use nuvocare/WikiMedical_sent_biobert with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("nuvocare/WikiMedical_sent_biobert") 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 nuvocare/WikiMedical_sent_biobert with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("nuvocare/WikiMedical_sent_biobert") model = AutoModel.from_pretrained("nuvocare/WikiMedical_sent_biobert", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,7 @@ datasets:
|
|
| 13 |
|
| 14 |
This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
|
| 15 |
|
| 16 |
-
WikiMedical_sent_bert is based on the [https://huggingface.co/dmis-lab/biobert-base-cased-v1.2](dmis-lab/biobert-base-cased-v1.2) backbone and has been trained on the [
|
| 17 |
|
| 18 |
The model is able to predict whether two texts are related to the same wikipedia page, with only medical topic.
|
| 19 |
|
|
|
|
| 13 |
|
| 14 |
This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
|
| 15 |
|
| 16 |
+
WikiMedical_sent_bert is based on the [https://huggingface.co/dmis-lab/biobert-base-cased-v1.2](dmis-lab/biobert-base-cased-v1.2) backbone and has been trained on the [WikiMedical_sentence_similarity](https://huggingface.co/datasets/nuvocare/WikiMedical_sentence_similarity) dataset.
|
| 17 |
|
| 18 |
The model is able to predict whether two texts are related to the same wikipedia page, with only medical topic.
|
| 19 |
|