How to use lion-ai/embeddinggemma-300m-medembed-triplets2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("lion-ai/embeddinggemma-300m-medembed-triplets2") sentences = [ "What were the results of the functional study using RNA and cDNA derivatives of the BRCA1 c.5074+3A>C variant?", "The patient was discharged on [date of discharge] and his most recent HbA1c level was 7.1% with a few episodes of hypoglycemia.", "The patient was diagnosed with stage IV NSCLC (malignant pleural effusion) in December 2014. Adenocarcinoma cells from pleural effusion were found, and immunohistochemistry analysis demonstrated positivity in TTF-1 and negativity in CK 5/6 and P63.", "Based on the results of the functional study using RNA and cDNA derivatives of the BRCA1 c.5074+3A>C variant, it is a likely pathogenic variant." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
