GPU_BRUTE_FORCE

Dedicado a ambientes de GPU, o índice GPU_BRUTE_FORCE foi concebido para cenários em que é essencial uma precisão sem compromissos. Garante uma recuperação de 1, comparando exaustivamente cada consulta com todos os vectores do conjunto de dados, assegurando que nenhuma correspondência potencial é ignorada. Aproveitando a aceleração da GPU, o GPU_BRUTE_FORCE é adequado para aplicações que exigem precisão absoluta em pesquisas de similaridade de vectores.

Criar índice

Para construir um índice GPU_BRUTE_FORCE num campo vetorial em Milvus, utilize o método add_index(), especificando os parâmetros index_type e metric_type para o í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
)

Nesta configuração:

• index_type: O tipo de índice a construir. Neste exemplo, defina o valor para GPU_BRUTE_FORCE.

• metric_type: O método utilizado para calcular a distância entre vectores. Para obter detalhes, consulte Tipos de métricas.

• params: Não são necessários parâmetros extra para o índice GPU_BRUTE_FORCE.

Assim que os parâmetros do índice estiverem configurados, pode criar o índice utilizando diretamente o método create_index() ou passando os parâmetros do índice no método create_collection. Para obter detalhes, consulte Criar coleção.

Pesquisar no índice

Assim que o índice for criado e as entidades forem inseridas, pode efetuar pesquisas de semelhança no í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 do índice

Para o índice GPU_BRUTE_FORCE, não são necessários parâmetros adicionais durante a criação do índice ou o processo de pesquisa.