DFP 训练模块
此模块的功能负责训练模型。
参数 |
类型 |
描述 |
示例值 |
默认值 |
|---|---|---|---|---|
feature_columns |
list |
要训练的特征列列表 | ["column1", "column2", "column3"] |
- |
epochs |
int |
训练轮次 | 50 |
- |
model_kwargs |
dict |
传递给模型的关键字参数 | {"encoder_layers": [64, 32], "decoder_layers": [32, 64], "activation": "relu", "swap_p": 0.1, "lr": 0.001, "lr_decay": 0.9, "batch_size": 32, "verbose": 1, "optimizer": "adam", "scalar": "min_max", "min_cats": 10, "progress_bar": false, "device": "cpu"} |
- |
validation_size |
float |
验证集的大小 | 0.1 |
- |
{
"feature_columns": [
"column1",
"column2",
"column3"
],
"epochs": 50,
"model_kwargs": {
"encoder_layers": [
64,
32
],
"decoder_layers": [
32,
64
],
"activation": "relu",
"swap_p": 0.1,
"lr": 0.001,
"lr_decay": 0.9,
"batch_size": 32,
"verbose": 1,
"optimizer": "adam",
"scalar": "min_max",
"min_cats": 10,
"progress_bar": false,
"device": "cpu"
},
"validation_size": 0.1
}