sentence-transformers How to use cooperzvegintzov/polyBERT with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("cooperzvegintzov/polyBERT")
sentences = [
"[*]CC[*]",
"[*]COC[*]",
"[*]CC(C)C[*]"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]
