回归模型¶
回归模型和CTR模型基本一致,只是采用的loss不一样。
如下图所示, 和CTR模型相比增加了: loss_type: L2_LOSS
model_config:{
model_class: "DeepFM"
feature_groups: {
group_name: "deep"
feature_names: "hour"
feature_names: "c1"
...
feature_names: "site_id_app_id"
wide_deep:DEEP
}
feature_groups: {
group_name: "wide"
feature_names: "hour"
feature_names: "c1"
...
feature_names: "c21"
wide_deep:WIDE
}
deepfm {
wide_output_dim: 16
dnn {
hidden_units: [128, 64, 32]
}
final_dnn {
hidden_units: [128, 64]
}
l2_regularization: 1e-5
}
embedding_regularization: 1e-7
loss_type: L2_LOSS
}