sentence-transformers How to use WeihaoLi/icd9 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("WeihaoLi/icd9")
sentences = [
"Complications of pregnancy; childbirth; and the puerperium → Complications during labor → Forceps delivery",
"Complications of pregnancy; childbirth; and the puerperium → Complications during labor",
"Complications of pregnancy; childbirth; and the puerperium → Other complications of birth; puerperium affecting management of mother",
"Complications of pregnancy; childbirth; and the puerperium → Normal pregnancy and/or delivery → Other pregnancy and delivery including normal"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
