内置算法的参数 - Amazon SageMaker

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

内置算法的参数

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

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

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

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

注意

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