hosa.optimization.hosa.HOSACNN.fit

HOSACNN.fit(max_gol_sizes, show_progress=True, imbalance_correction=None, **kwargs)[source]

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

Parameters

  • max_gol_sizes (int) – Maximum number of GofLayers 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.

  • imbalance_correction – 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.