亚马逊 SageMaker 模板 - 亚马逊 OpenSearch 服务

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

亚马逊 SageMaker 模板

Amazon SageMaker CloudFormation 模板定义了多种 AWS 资源,以便为您设置神经插件和语义搜索。

首先,使用 Amazon SageMaker 模板与文本嵌入模型集成,在 SageMaker 运行时中将文本嵌入模型部署为服务器。如果您不提供模型终端节点,则 CloudFormation 创建一个 IAM 角色以允许 SageMaker Runtime 从 Amazon S3 下载模型工件并将其部署到服务器。如果您提供终端节点,则 CloudFormation 会创建一个允许 Lambda 函数访问 OpenSearch 服务域的 IAM 角色,或者,如果该角色已经存在,则更新和重复使用该角色。该端点通过 ML Commons 插件为用于 ML 连接器的远程模型提供服务。

然后,使用 “通过 Amazon 与稀疏编码器集成” SageMaker 模板创建 Lambda 函数,让您的域设置远程推理连接器。在 S OpenSearch ervice 中创建连接器后,远程推理可以在 Runtime 中 SageMaker 使用远程模型运行语义搜索。该模板会将域中的模型 ID 返回给您,以便您可以开始搜索。

使用 Amazon SageMaker AI CloudFormation 模板
  1. 打开亚马逊 OpenSearch 服务控制台

  2. 在左侧导航窗格中,选择集成。

  3. 在每个 Amazon SageMaker AI 模板下,选择配置域配置公共域

  4. 按照 CloudFormation 控制台中的提示配置堆栈并设置模型。

注意

OpenSearch 服务还提供了一个单独的模板来配置 VPC 域。如果您使用此模板,则需要提供 Lambda 函数的 VPC ID。