How to use WeihaoLi/full_icd with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("WeihaoLi/full_icd") sentences = [ "Infectious and parasitic diseases → Viral infection", "Diseases of the blood and blood-forming organs", "Infectious and parasitic diseases", "Diseases of the nervous system and sense organs → Central nervous system infection" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
