sentence-transformers How to use blachang28/balanced-amc-gemma with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("blachang28/balanced-amc-gemma")
sentences = [
"Let $n$ be the least positive integer greater than $1000$ for which \\[\\gcd(63, n+120) =21\\quad \\text{and} \\quad \\gcd(n+63, 120)=60.\\] What is the sum of the digits of $n$?",
"Jason rolls three fair standard six-sided dice. Then he looks at the rolls and chooses a subset of the dice (possibly empty, possibly all three dice) to reroll. After rerolling, he wins if and only if the sum of the numbers face up on the three dice is exactly $7.$ Jason always plays to optimize his chances of winning. What is the probability that he chooses to reroll exactly two of the dice?",
"$2000(2000^{2000}) = x$ Find x.",
"An $8$-foot by $10$-foot bathroom floor is tiled with square tiles of size $1$ foot by $1$ foot. Each tile has a pattern consisting of four white quarter circles of radius $1/2$ foot centered at each corner of the tile. The remaining portion of the tile is shaded. How many square feet of the floor are shaded? [asy] unitsize(2cm); defaultpen(linewidth(.8pt)); fill(unitsquare,gray); filldraw(Arc((0,0),.5,0,90)--(0,0)--cycle,white,black); filldraw(Arc((1,0),.5,90,180)--(1,0)--cycle,white,black); filldraw(Arc((1,1),.5,180,270)--(1,1)--cycle,white,black); filldraw(Arc((0,1),.5,270,360)--(0,1)--cycle,white,black); [/asy]"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
