GPU_BRUTE_FORCE

Dedicato agli ambienti GPU, l'indice GPU_BRUTE_FORCE è stato progettato per scenari in cui è essenziale una precisione senza compromessi. Garantisce un richiamo di 1 confrontando in modo esaustivo ogni query con tutti i vettori del set di dati, assicurando che nessuna potenziale corrispondenza venga trascurata. Sfruttando l'accelerazione delle GPU, GPU_BRUTE_FORCE è adatto alle applicazioni che richiedono una precisione assoluta nelle ricerche di similarità vettoriale.

Creazione dell'indice

Per costruire un indice GPU_BRUTE_FORCE su un campo vettoriale in Milvus, utilizzare il metodo add_index(), specificando i parametri index_type e metric_type per l'indice.

from pymilvus import MilvusClient

# Prepare index building params
index_params = MilvusClient.prepare_index_params()

index_params.add_index(
    field_name="your_vector_field_name", # Name of the vector field to be indexed
    index_type="GPU_BRUTE_FORCE", # Type of the index to create
    index_name="vector_index", # Name of the index to create
    metric_type="L2", # Metric type used to measure similarity
    params={} # No additional parameters required for GPU_BRUTE_FORCE
)

In questa configurazione:

• index_type: Il tipo di indice da costruire. In questo esempio, impostare il valore su GPU_BRUTE_FORCE.

• metric_type: Il metodo utilizzato per calcolare la distanza tra i vettori. Per i dettagli, fare riferimento a Tipi metrici.

• params: Per l'indice GPU_BRUTE_FORCE non sono necessari altri parametri.

Una volta configurati i parametri dell'indice, è possibile creare l'indice utilizzando direttamente il metodo create_index() o passando i parametri dell'indice al metodo create_collection. Per i dettagli, fare riferimento a Creare una raccolta.

Ricerca nell'indice

Una volta creato l'indice e inserite le entità, è possibile eseguire ricerche di similarità sull'indice.

res = MilvusClient.search(
    collection_name="your_collection_name", # Collection name
    anns_field="vector_field", # Vector field name
    data=[[0.1, 0.2, 0.3, 0.4, 0.5]],  # Query vector
    limit=3,  # TopK results to return
    search_params={"params": {}}  # No additional parameters required for GPU_BRUTE_FORCE
)

Parametri dell'indice

Per l'indice GPU_BRUTE_FORCE non sono necessari parametri aggiuntivi né durante la creazione dell'indice né durante il processo di ricerca.