此版本的文档已存档,不再提供支持。查看 最新文档,学习;了解如何升级您的Java驾驶员版本。
Overview
在本指南中,您可以学习如何使用 Atlas Vector Search 功能,适用于 Java 驱动程序。The Aggregates
构建者 类 提供了 vectorSearch()
辅助工具 方法,您可以用它来创建 $vectorSearch 管道 阶段。此管道阶段允许您对文档进行语义搜索。语义搜索是一种搜索类型,可以查找与您提供的搜索词或短语在含义上相似但不完全相同的信息。
执行向量搜索
如要使用此功能,您必须创建向量搜索索引并为向量嵌入创建索引。如要了解如何以编程方式创建向量搜索索引,请参阅索引指南中的 Atlas Search 索引部分。如要了解有关向量嵌入的更多信息,请参阅 Atlas 文档中的如何为向量搜索的向量嵌入创建索引。
在向量嵌入上创建向量搜索索引后,您可以在管道阶段引用此索引,如下节所示。
向量搜索示例
以下示例展示了如何构建一个聚合管道,从而使用 vectorSearch()
和 project()
方法计算向量搜索分数:
List<Double> queryVector = (asList(-0.0072121937, -0.030757688, -0.012945653)); String indexName = "mflix_movies_embedding_index"; FieldSearchPath fieldSearchPath = fieldPath("plot_embedding"); int numCandidates = 2; int limit = 1; VectorSearchOptions options = vectorSearchOptions().filter(gte("year", 2016)); List<Bson> pipeline = asList( vectorSearch( fieldSearchPath, queryVector, indexName, numCandidates, limit, options), project( metaVectorSearchScore("vectorSearchScore")));
以下示例展示了如何运行聚合并从上述聚合管道的结果中打印向量搜索分数:
Document found = collection.aggregate(pipeline).first(); double score = found.getDouble("vectorSearchScore").doubleValue(); System.out.println("vectorSearch score: " + score);
API 文档
要进一步了解本指南所提及的方法和类型,请参阅以下 API 文档: