MLP

class lightautoml.ml_algo.torch_based.nn_models.MLP(*args, **kwargs)[source]

Bases: DenseLightModel

Realisation of ‘mlp’ model.

Parameters:

  • n_in – Input dim.

  • n_out – Output dim.

  • hidden_size – List of hidden dims.

  • drop_rate – Dropout rate for each layer separately or altogether.

  • act_fun – Activation function.

  • noise_std – Std of noise.

  • num_init_features – If not none add fc layer before model with certain dim.

  • use_bn – Use BatchNorm.

  • use_noise – Use noise.

  • device – Device to compute on.