modnet.model_presets package


Module contents

This submodule defines some MODNet model presets which are used in the fit_preset function.

modnet.model_presets.gen_presets(n_feat, n_samples, classification=False)

Generates sensible preset architectures and learning parameters based on number of samples and features.

  • n_feat (int) – The number of training features available to the model.

  • n_samples (int) – The number of training samples available to the model.

  • classification (bool) –


List of dictionaries to individually pass as kwargs to

Return type:

List[Dict[str, Any]]