GPU_BRUTE_FORCE

Предназначенный для сред на GPU, индекс GPU_BRUTE_FORCE разработан для сценариев, в которых важна бескомпромиссная точность. Он гарантирует отзыв 1 за счет исчерпывающего сравнения каждого запроса со всеми векторами в наборе данных, гарантируя, что ни одно потенциальное совпадение не будет упущено. Используя ускорение GPU, GPU_BRUTE_FORCE подходит для приложений, требующих абсолютной точности при поиске векторного сходства.

Построение индекса

Чтобы построить индекс GPU_BRUTE_FORCE по векторному полю в Milvus, используйте метод add_index(), указав параметры index_type и metric_type для индекса.

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
)

В данной конфигурации:

• index_type: Тип индекса, который будет построен. В этом примере задайте значение GPU_BRUTE_FORCE.

• metric_type: : Метод, используемый для вычисления расстояния между векторами. Подробнее см. в разделе Метрические типы.

• params: Для индекса GPU_BRUTE_FORCE дополнительные параметры не требуются.

После того как параметры индекса настроены, вы можете создать индекс, используя метод create_index() напрямую или передавая параметры индекса в метод create_collection. Подробнее см. в разделе Создание коллекции.

Поиск по индексу

После того как индекс создан и сущности вставлены в него, можно выполнять поиск по сходству в индексе.

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
)

Параметры индекса

Для индекса GPU_BRUTE_FORCE не требуется никаких дополнительных параметров ни при создании индекса, ни в процессе поиска.