本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
调整 DeepAR 模型
自动模型优化(也称作超参数优化)通过运行很多在数据集上测试一系列超参数的作业来查找模型的最佳版本。您可以选择可优化超参数、每个超参数的值范围和一个目标指标。您可以从算法计算的指标中选择目标指标。自动模型优化将搜索所选超参数以找到导致优化目标指标的模型的值组合。
有关模型优化的更多信息,请参阅自动调整模型 SageMaker。
DeepAR 算法计算的指标
DeepAR 算法报告在训练期间计算的三个指标。在调整模型时,请选择这些指标之一作为目标指标。对于目标,请使用所提供测试通道上的预测准确性(推荐)或训练损失。有关 DeepAR 算法的训练/测试拆分的建议,请参阅 使用 DeepAR 算法的最佳实践。
指标名称 | 描述 | 优化方向 |
---|---|---|
test:RMSE |
在测试集上计算的预测与实际目标之间的均方根误差。 |
最小化 |
test:mean_wQuantileLoss |
在测试集上计算出的平均总体分位数损失。要控制使用什么分位数,请设置 |
最小化 |
train:final_loss |
训练负对数似然损失,对模型在上一个训练纪元取平均值。 |
最小化 |
DeepAR 算法的可调整超参数。
使用以下超参数调整 DeepAR 模型。对 DeepAR 目标指标产生最大影响的超参数(从影响最大到最小的顺序列出)包括:epochs
、context_length
、mini_batch_size
、learning_rate
和 num_cells
。
参数名称 | 参数类型 | 建议的范围 |
---|---|---|
epochs |
|
MinValue: 1, MaxValue: 1000 |
context_length |
|
MinValue: 1, MaxValue: 200 |
mini_batch_size |
|
MinValue: 32, MaxValue: 1028 |
learning_rate |
|
MinValue: 1e-5, MaxValue: 1e-1 |
num_cells |
|
MinValue: 30, MaxValue: 200 |
num_layers |
|
MinValue: 1, MaxValue: 8 |
dropout_rate |
|
MinValue: 0.00, MaxValue: 0.2 |
embedding_dimension |
|
MinValue: 1, MaxValue: 50 |