17 lines
649 B
Python
17 lines
649 B
Python
from sentence_transformers import SentenceTransformer
|
|
# Preguntas y respuestas especializado en eso "multi-qa-mpnet-base-dot-v1"
|
|
# uno de uso gereal el de mejor desempeño all-mpnet-base-v2
|
|
# el mas rapido "paraphrase-MiniLM-L3-v2" y "all-MiniLM-L6-v2"
|
|
# muy rappudo y muy acertado "all-MiniLM-L12-v2"
|
|
#models=["all-MiniLM-L12-v2","paraphrase-MiniLM-L3-v2" , "all-MiniLM-L6-v2",
|
|
models=["all-mpnet-base-v2","multi-qa-mpnet-base-dot-v1"]
|
|
|
|
for model in models:
|
|
modelST = SentenceTransformer(model)
|
|
# Define the path where you want to save the model
|
|
save_path = './embeddings/%s/'%(model)
|
|
# Save the model
|
|
modelST.save(save_path)
|
|
|
|
|