调整因子分解机模型
自动模型优化（也称作超参数优化）通过运行很多在数据集上测试一系列超参数的作业来查找模型的最佳版本。您可以选择可优化超参数、每个超参数的值范围和一个目标指标。您可以从算法计算的指标中选择目标指标。自动模型优化将搜索所选超参数以找到导致优化目标指标的模型的值组合。
有关模型优化的更多信息，请参阅使用 SageMaker 执行自动模型调优。
由因子分解机算法计算的指标
因子分解机算法具有二元分类和回归预测器类型。预测器类型确定可用于自动模型优化的指标。该算法报告在训练期间计算的 test:rmse 回归量指标。在为回归任务优化模型时，请选择此指标作为目标。
| 指标名称 | 描述 | 优化方向 |
|---|---|---|
test:rmse |
均方根误差 |
最小化 |
因子分解机算法报告在训练期间计算的 3 个二元分类指标。在为二元分类任务优化模型时，请选择其中之一作为目标。
| 指标名称 | 描述 | 优化方向 |
|---|---|---|
test:binary_classification_accuracy |
准确性 |
最大化 |
test:binary_classification_cross_entropy |
交叉熵 |
最小化 |
test:binary_f_beta |
测试版 |
最大化 |
可优化因子分解机超参数
您可为因子分解机算法调整以下超参数。包含偏置项、线性项和因子分解项的初始化参数取决于其初始化方法。有三种初始化方法：uniform、normal 和 constant。这些初始化方法本身是不可优化的。可优化参数依赖于初始化方法的选择。例如，如果初始化方法为 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 |
无 |