File size: 448 Bytes
80b95e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("google/embeddinggemma-300m")
query = "Which planet is known as the Red Planet?"
docs = [
"Mars, known for its reddish appearance, is often referred to as the Red Planet.",
"Venus is similar in size to Earth.",
]
import numpy as np
q_emb = np.array(model.encode_query(query))
d_emb = np.array(model.encode_document(docs))
print(q_emb.shape, d_emb.shape)
|