DFP Training Module - NVIDIA Docs

This module function is responsible for training the model.

Configurable Parameters

Parameter	Type	Description	Example Value	Default Value
feature_columns	list	List of feature columns to train on	[“column1”, “column2”, “column3”]	`-`
epochs	int	Number of epochs to train for	50	`-`
model_kwargs	dict	Keyword arguments to pass to the model	{“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	Size of the validation set	0.1	`-`

JSON Example

Copy
Copied!

            
            {
  "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
}