modnet.model_presets package

Submodules

• modnet.model_presets.presets module
  • gen_presets()

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.

Parameters:

• 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) –

Returns:

List of dictionaries to individually pass as kwargs to model.fit(...).

Return type:

List[Dict[str, Any]]