How to use albertus-sussex/veriscrape-book-test-sbert-bs128_lr0.0002_ep5_cosine_snTrue_spFalse_hn1 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("albertus-sussex/veriscrape-book-test-sbert-bs128_lr0.0002_ep5_cosine_snTrue_spFalse_hn1", trust_remote_code=True) sentences = [ "Don Piper", "Tommy Nelson", "Kate Walbert", "publisher", "author" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [5, 5]
