OptunaMixin

class falcon.abstract.OptunaMixin

Abstract mixin that should be used in order to indicate the compatibility of the model with OptunaLearner.

abstract classmethod get_search_space(X: Any, y: Any) Union[Callable, Dict]

A class method that provides an optuna search space for the model. Optionally, the search space can be adjusted based on the provided training data.

Parameters
  • X (Any) – features

  • y (Any) – targets

Returns

dictionary that describes the search space, or custom objective function

Return type

Union[Callable, Dict]