hosa.optimization.hosa.HOSARNN.fit

HOSARNN.fit(max_n_subs_layers, show_progress=True, imbalance_correction=None, **kwargs)[source]

Optimize the model following the HOSA approach with all sets of parameters.

Parameters

  • max_n_subs_layers (int) – Maximum number of subsequent layers to add to the model.

  • show_progress (bool) – True to show a progress bar; False otherwise.

  • imbalance_correction (None or bool) – Whether to apply correction to class imbalances. Only used for classification problems. Ignored for regression.

  • **kwargs – Extra arguments explicitly used for regression or classification models, including the additional arguments that are used in the TensorFlow’s model fit function. See here.

Returns

tuple – Returns a tuple containing the object of the best model found and respective performance metric and optimized parameters.