sentence-transformers How to use ksg-dfci/TrialSpace-0526 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("ksg-dfci/TrialSpace-0526")
sentences = [
"Instruct: Given a cancer patient summary, retrieve clinical trial options that are reasonable for that patient; or, given a clinical trial option, retrieve cancer patients who are reasonable candidates for that trial. Age: 39\nSex: Female\nCancer type: T-cell acute lymphoblastic leukemia (T-ALL)\nHistology: T-cell lymphoblasts\nCurrent extent: Advanced (bone marrow, peripheral blood, cervical lymph nodes, and spleen)\nBiomarkers: CD2 positive, CD3 positive (cytoplasmic and surface), CD5 positive, CD7 positive, NOTCH1 p.T1851I, PTEN homozygous loss, TP53 p.R175H, SETD2 p.V123L\nTreatment history:\n2015-09-18 to 2017-02-11: vincristine, prednisone, and cytarabine; best response complete remission\n2017: allogeneic hematopoietic stem cell transplant\n2019-06-14 to 2020-11-09: cladribine; best response partial response (bone marrow blasts decreased from 12 percent to 6 percent), now refractory",
"Instruct: Given a cancer patient summary, retrieve clinical trial options that are reasonable for that patient; or, given a clinical trial option, retrieve cancer patients who are reasonable candidates for that trial. Age range allowed: 0 to 6 years. Sex allowed: Both. Cancer type allowed: Central nervous system cancer. Histology allowed: Group 3 medulloblastoma. Cancer burden allowed: Metastatic disease allowed. Prior treatment required: NA. Prior treatment excluded: Radiotherapy and chemotherapy (excluding corticosteroids). Biomarkers required: NA. Biomarkers excluded: NA.",
"Instruct: Given a cancer patient summary, retrieve clinical trial options that are reasonable for that patient; or, given a clinical trial option, retrieve cancer patients who are reasonable candidates for that trial. Age range allowed: 0 to 6 years. Sex allowed: Both. Cancer type allowed: Central nervous system cancer. Histology allowed: Central nervous system embryonal tumor with rhabdoid features. Cancer burden allowed: Metastatic disease allowed. Prior treatment required: NA. Prior treatment excluded: Radiotherapy and chemotherapy (excluding corticosteroids). Biomarkers required: INI-1 intact. Biomarkers excluded: NA.",
"Instruct: Given a cancer patient summary, retrieve clinical trial options that are reasonable for that patient; or, given a clinical trial option, retrieve cancer patients who are reasonable candidates for that trial. Age range allowed: 0 to 29 years. Sex allowed: both. Cancer type allowed: T-cell lymphoma. Histology allowed: T-cell lymphoma. Cancer burden allowed: Second or greater relapse or first relapse post-allogeneic hematopoietic stem cell transplant or primary refractory disease or relapsed and refractory disease. Prior treatment required: NA. Prior treatment excluded: CD7 targeted therapy. Biomarkers required: CD7 expression on T-cell lymphoma blasts (at least 90% of blasts positive by flow cytometry or immunohistochemistry) assessed during screening. Biomarkers excluded: NA."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
