调整因子分解机模型 - Amazon SageMaker
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

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

调整因子分解机模型

自动模型优化(也称作超参数优化)通过运行很多在数据集上测试一系列超参数的作业来查找模型的最佳版本。您可以选择可优化超参数、每个超参数的值范围和一个目标指标。您可以从算法计算的指标中选择目标指标。自动模型优化将搜索所选超参数以找到导致优化目标指标的模型的值组合。

有关模型优化的更多信息,请参阅执行自动模型优化

由因子分解机算法计算的指标

因子分解机算法具有二进制分类和回归预测器类型。预测器类型确定可用于自动模型优化的指标。该算法报告在训练期间计算的 test:rmse 回归量指标。在为回归任务优化模型时,请选择此指标作为目标。

指标名称 描述 优化方向
test:rmse

均方根误差

最小化

因子分解机算法报告三个二进制分类指标,这些指标在训练期间计算。在为二元分类任务优化模型时,请选择其中之一作为目标。

指标名称 描述 优化方向
test:binary_classification_accuracy

Accuracy

最大化

test:binary_classification_cross_entropy

交叉熵

最小化

test:binary_f_beta

测试版

最大化

可优化因子分解机超参数

您可以为因子分解机算法优化以下超参数。包含偏置项、线性项和因子分解项的初始化参数取决于其初始化方法。有三种初始化方法:uniformnormalconstant。这些初始化方法本身是不可优化的。可优化参数依赖于初始化方法的选择。例如,如果初始化方法为 uniform,则仅 scale 参数是可优化的。具体而言,如果 bias_init_method==uniform,则 bias_init_scale、 和 linear_init_scale factors_init_scale 是可优化的。同样,如果初始化方法为 normal,则仅 sigma 参数是可优化的。如果初始化方法为 constant,则仅 value 参数是可优化的。下表列出了这些依赖项。

参数名称 参数类型 建议的范围 依赖关系
bias_init_scale

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==uniform

bias_init_sigma

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==normal

bias_init_value

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==constant

bias_lr

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_wd

ContinuousParameterRange

最小值:1e-8,最大值:512

epoch

IntegerParameterRange

最小值:1,最大值:1000

factors_init_scale

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==uniform

factors_init_sigma

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==normal

factors_init_value

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==constant

factors_lr

ContinuousParameterRange

最小值:1e-8,最大值:512

factors_wd

ContinuousParameterRange

最小值:1e-8,最大值:512]

linear_init_scale

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==uniform

linear_init_sigma

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==normal

linear_init_value

ContinuousParameterRange

最小值:1e-8,最大值:512

bias_init_method==constant

linear_lr

ContinuousParameterRange

最小值:1e-8,最大值:512

linear_wd

ContinuousParameterRange

最小值:1e-8,最大值:512

mini_batch_size

IntegerParameterRange

最小值:100,最大值:10000