Snowflake / snowflake-arctic-embed-l-v2.0
Milvus Integrated
Tâche: Intégration
Modalité: Texte
Métrique de similarité: IP, L1, L2, cosinus
Licence: Apache 2.0
Dimensions: 1024
Tokens d'entrée max: 8192
Prix: Gratuit
Introduction à snowflake-arctic-embed-l-v2.0
Le modèle snowflake-arctic-embed-l-v2.0 est un modèle multilingue d’embeddings de texte prenant en charge 74 langues, conçu pour fournir une récupération de haute qualité et une inférence efficace pour les tâches en anglais comme dans d’autres langues. Avec 303M de paramètres hors embeddings, il offre une inférence rapide et évolutive, et prend en charge des représentations compactes grâce à Matryoshka Representation Learning et à l’entraînement tenant compte de la quantification.
Comparez snowflake-arctic-embed-l-v2.0 avec les modèles snowflake-arctic-l :
| Modèle | params | params hors emb | dimensions | Prise en charge linguistique |
|---|---|---|---|---|
| snowflake-arctic-l-v2.0 | 568M | 303M | 1024 | multilingue |
| snowflake-arctic-l | 335M | 303M | 1024 | anglais uniquement |
Comment créer des embeddings avec snowflake-arctic-embed-l-v2.0
Il existe deux méthodes principales pour générer des embeddings vectoriels :
- PyMilvus : le SDK Python pour Milvus qui intègre de manière transparente le modèle
snowflake-arctic-embed-l-v2.0. - La fonction
AI_EMBEDfournie par Snowflake Cortex pour le texte et les images.
Une fois les embeddings vectoriels générés, ils peuvent être stockés dans Zilliz Cloud (un service de base de données vectorielle entièrement géré, propulsé par Milvus) et utilisés pour la recherche par similarité sémantique. Voici quatre étapes clés :
- Inscrivez-vous gratuitement pour obtenir un compte Zilliz Cloud.
- Configurez un cluster serverless et obtenez le Public Endpoint et l’API Key.
- Créez une collection vectorielle et insérez vos embeddings vectoriels.
- Exécutez une recherche sémantique sur les embeddings stockés.
Créer des embeddings via PyMilvus et les insérer dans Zilliz Cloud pour la recherche sémantique
from pymilvus.model.dense import SentenceTransformerEmbeddingFunction
from pymilvus import MilvusClient
# Load the Snowflake Arctic Embed L v2.0 model
ef = SentenceTransformerEmbeddingFunction(
"Snowflake/snowflake-arctic-embed-l-v2.0", trust_remote_code=True
)
docs = [
"Artificial intelligence was founded as an academic discipline in 1956.",
"Alan Turing was the first person to conduct substantial research in AI.",
"Born in Maida Vale, London, Turing was raised in southern England.",
]
# Generate embeddings for documents
docs_embeddings = ef(docs)
queries = ["When was artificial intelligence founded", "Where was Alan Turing born?"]
# Generate embeddings for queries
query_embeddings = ef(queries)
# Connect to Zilliz Cloud with Public Endpoint and API Key
client = MilvusClient(uri=ZILLIZ_PUBLIC_ENDPOINT, token=ZILLIZ_API_KEY)
COLLECTION = "arctic_embed_l_v2_documents"
# Drop collection if it exists
if client.has_collection(collection_name=COLLECTION):
client.drop_collection(collection_name=COLLECTION)
# Create collection with auto-detected dimension
client.create_collection(collection_name=COLLECTION, dimension=ef.dim, auto_id=True)
# Insert documents with embeddings
for doc, embedding in zip(docs, docs_embeddings):
client.insert(COLLECTION, {"text": doc, "vector": embedding})
# Search for similar documents
results = client.search(
collection_name=COLLECTION,
data=query_embeddings,
# consistency_level="Strong", # Strong consistency ensures accurate results but may increase latency
output_fields=["text"],
limit=2,
)
# Print search results
for i, query in enumerate(queries):
print(f"\nQuery: {query}")
for result in results[i]:
print(f" - {result['entity']['text']} (distance: {result['distance']:.4f})")
Pour plus d’informations, consultez notre documentation du modèle d’intégration PyMilvus.
- Introduction à snowflake-arctic-embed-l-v2.0
- Comment créer des embeddings avec snowflake-arctic-embed-l-v2.0
Contenu
Workflows IA fluides
Des embeddings à la recherche IA évolutive - Zilliz Cloud vous permet de stocker, indexer et récupérer des embeddings avec une vitesse et une efficacité inégalées.
Essayer Zilliz Cloud gratuitement
