sentence-transformers How to use RK-1235/bge-base-FIR-matryoshka-BASELINE-10epochs-FT with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("RK-1235/bge-base-FIR-matryoshka-BASELINE-10epochs-FT")
sentences = [
"Employee health, safety and wellness are top priorities at Hasbro. We support our colleagues’ well-being, which includes mental, physical and financial wellness, through a number of programs, including: robust employee assistance programs, childcare solutions, and a commitment to flexible work arrangements.",
"What percentage of the total annual net trade sales did the sales returns reserve represent for the company during each of the fiscal years 2023, 2022, and 2021?",
"How does Hasbro support the wellness of its employees?",
"What was the conclusion of the Company's review regarding the impact of the American Rescue Plan, the Consolidated Appropriations Act, 2021, and related tax provisions on its business for the fiscal year ended June 30, 2023?"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]