調校 Factorization Machines 模型 - Amazon SageMaker

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

調校 Factorization Machines 模型

自動模型調校,又稱為超參數調校,會透過在您的資料集上,執行許多測試超參數範圍的任務,來尋找最佳版本的模型。您可以選擇可調校的超參數、每一個超參數的值範圍,及目標指標。您可以從演算法運算的指標中選擇目標指標。自動模型調校會搜尋所選擇的超參數,以找出產生之模型可最佳化目標指標的值組合。

如需模型調校的詳細資訊,請參閱使用 自動模型調校 SageMaker

依 Factorization Machines 演算法運算的指標

Factorization Machines 演算法具有二元分類和迴歸預測器兩種類型。預測器類型會判斷可用於自動模型調校的指標。演算法會回報 test:rmse 迴歸指標 (在訓練期間運算)。調校迴歸任務的模型時,請選擇此指標做為目標。

指標名稱 描述 最佳化方向
test:rmse

均方根誤差

最小化

Factorization Machines 算法會回報三個二元分類指標 (在訓練期間運算)。調校二元分類任務的模型時,請選擇其中之一做為目標。

指標名稱 描述 最佳化方向
test:binary_classification_accuracy

準確性

最大化

test:binary_classification_cross_entropy

跨熵

最小化

test:binary_f_beta

試用版

最大化

可調校的 Factorization Machines 超參數

您可以調校因式分解機演算法的下列超參數。包含項偏差、線性和因式分解的初始化參數,取決於其初始化方法。有三種初始化方法:uniformnormalconstant。這些初始化方法本身並不可調校。可調校參數取決於初始化方法的此選擇。例如,如果初始化方法是 uniform,則只有 scale 參數可調校。具體而言,如果 bias_init_method==uniform,則 bias_init_scalelinear_init_scalefactors_init_scale 可調校。同樣,如果初始化方法是 normal,則只有 sigma 參數可調校。如果初始化方法是 constant,則只有 value 參數可調校。下表列出這些相依性。

參數名稱 參數類型 建議範圍 相依性
bias_init_scale

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==uniform

bias_init_sigma

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==normal

bias_init_value

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==constant

bias_lr

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_wd

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

epoch

IntegerParameterRange

MinValue:1, MaxValue:1000

factors_init_scale

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==uniform

factors_init_sigma

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==normal

factors_init_value

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==constant

factors_lr

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

factors_wd

ContinuousParameterRange

MinValue:1e-8, MaxValue:512】

linear_init_scale

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==uniform

linear_init_sigma

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==normal

linear_init_value

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

bias_init_method==constant

linear_lr

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

linear_wd

ContinuousParameterRange

MinValue:1e-8, MaxValue:512

mini_batch_size

IntegerParameterRange

MinValue:100, MaxValue:10000