GPU_BRUTE_FORCE

Dedicado a entornos de GPU, el índice GPU_BRUTE_FORCE está diseñado para escenarios en los que es esencial una precisión sin concesiones. Garantiza una recuperación de 1 comparando exhaustivamente cada consulta con todos los vectores del conjunto de datos, lo que garantiza que no se pase por alto ninguna posible coincidencia. Aprovechando la aceleración de la GPU, GPU_BRUTE_FORCE es adecuado para aplicaciones que exigen una precisión absoluta en las búsquedas de similitud vectorial.

Construir índice

Para construir un índice GPU_BRUTE_FORCE sobre un campo vectorial en Milvus, utilice el método add_index(), especificando los parámetros index_type y metric_type para el índice.

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
)

En esta configuración:

• index_type: El tipo de índice a construir. En este ejemplo, establezca el valor GPU_BRUTE_FORCE.

• metric_type: El método utilizado para calcular la distancia entre vectores. Para más detalles, consulte Tipos métricos.

• params: No se necesitan parámetros adicionales para el índice GPU_BRUTE_FORCE.

Una vez configurados los parámetros del índice, puedes crear el índice utilizando el método create_index() directamente o pasando los parámetros del índice en el método create_collection. Para más detalles, consulte Crear colección.

Búsqueda en el índice

Una vez creado el índice e insertadas las entidades, puede realizar búsquedas de similitud en el índice.

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
)

Parámetros del índice

Para el índice GPU_BRUTE_FORCE, no se necesitan parámetros adicionales ni durante la creación del índice ni durante el proceso de búsqueda.