How to use jo-mengr/mmcontext-pubmedbert-gs10k with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("jo-mengr/mmcontext-pubmedbert-gs10k") sentences = [ "sample_idx:SRX956540", "This measurement was conducted with Illumina HiSeq 2500. Primary neutrophil cells obtained from a healthy adult male.", "This measurement was conducted with Illumina HiSeq 2000. 1000 ng of fragmented total RNA from a cultured chronic myelogenous leukemia (CML) cell line (K-562) derived from a female hematological system disease (CML). The cells were grown in tissue culture and have undergone Ribo-Zero treatment.", "sample_idx:SRX185895" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
