内置算法的参数 - 亚马逊 SageMaker AI

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

内置算法的参数

下表列出了 Amazon A SageMaker I 提供的每种算法的参数。

算法名称 渠道名称 训练输入模式 文件类型 实例类 可并行化
AutoGluon-表格 训练和(可选)验证 文件 CSV CPU或GPU(仅限单实例)
BlazingText 训练 文件或管道 文本文件(每行一句,带空格分隔的令牌) CPU或GPU(仅限单实例)
CatBoost 训练和(可选)验证 文件 CSV CPU(仅限单实例)
DeepAR 预测 训练和 (可选) 测试 文件 JSON线条或实木复合地板 CPU 或 GPU Yes
因子分解机 训练和 (可选) 测试 文件或管道 recordIO-protobuf CPU(GPU适用于密集数据) Yes
图像分类-MXNet 训练和验证,(可选)train_lst、validation_lst 和模型 文件或管道 recordIO 或图像文件 (.jpg 或 .png) GPU Yes
图像分类- TensorFlow 训练和验证 文件 图像文件(.jpg、.jpeg 或 .png) CPU 或 GPU 是(仅在单个实例GPUs上跨多个实例)
IP 见解 训练和 (可选) 验证 文件 CSV CPU 或 GPU Yes
K-Means 训练和 (可选) 测试 文件或管道 recordio-protobuf 或 CSV CPU或GPUCommon(一个或多个实例上的单个GPU设备)
K-Nearest-Neighbors (k-nn) 训练和 (可选) 测试 文件或管道 recordio-protobuf 或 CSV CPU或GPU(一个或多个实例上的单个GPU设备) Yes
LDA 训练和 (可选) 测试 文件或管道 recordio-protobuf 或 CSV CPU(仅限单实例)
光 GBM 训练和(可选)验证 文件 CSV CPU Yes
线性学习器 训练和 (可选) 验证和/或测试 文件或管道 recordio-protobuf 或 CSV CPU 或 GPU Yes
神经主题模型 训练和 (可选) 验证和/或测试 文件或管道 recordio-protobuf 或 CSV CPU 或 GPU Yes
Object2Vec 训练和 (可选) 验证和/或测试 文件 JSON台词 CPU或GPU(仅限单实例)
物体检测-MXNet 训练和验证,(可选)train_annotation、validation_annotation 和模型 文件或管道 recordIO 或图像文件 (.jpg 或 .png) GPU Yes
物体检测- TensorFlow 训练和验证 文件 图像文件(.jpg、.jpeg 或 .png) GPU 是(仅在单个实例GPUs上跨多个实例)
PCA 训练和 (可选) 测试 文件或管道 recordio-protobuf 或 CSV CPU 或 GPU Yes
Random Cut Forest 训练和 (可选) 测试 文件或管道 recordio-protobuf 或 CSV CPU Yes
语义分割 训练和验证、train_annotation、validation_annotation 以及(可选)label_map 和模型 文件或管道 图像文件 GPU(仅限单实例)
Seq2Seq 建模 训练、验证和 vocab 文件 recordIO-protobuf GPU(仅限单实例)
TabTransformer 训练和(可选)验证 文件 CSV CPU或GPU(仅限单实例)
文本分类- TensorFlow 训练和验证 文件 CSV CPU 或 GPU 是(仅在单个实例GPUs上跨多个实例)
XGBoost(0.90-1、0.90-2、1.0-1、1.2-1、1.2-21) 训练和 (可选) 验证 文件或管道 CSV、Lib 或 Par SVM quet CPU(或者GPU适用于 1.2-1) Yes

可并行化 的算法可部署在多个计算实例上以进行分布式训练。

以下主题提供有关数据格式、推荐的 Amazon EC2 实例类型以及 Amazon A SageMaker I 提供的所有内置算法的通用 CloudWatch 日志的信息。

注意

要查找 SageMaker AI 管理URIs的内置算法的 Docker 镜像,请参阅 Docker 注册表路径和示例代码。