evalml.automl.automl_algorithm.AutoMLAlgorithm.__init__

AutoMLAlgorithm.__init__(allowed_pipelines=None, max_pipelines=None, tuner_class=None, random_state=0)[source]

This class represents an automated machine learning (AutoML) algorithm. It encapsulates the decision-making logic behind an automl search, by both deciding which pipelines to evaluate next and by deciding what set of parameters to configure the pipeline with.

To use this interface, you must define a next_batch method which returns the next group of pipelines to evaluate on the training data. That method may access state and results recorded from the previous batches, although that information is not tracked in a general way in this base class. Overriding add_result is a convenient way to record pipeline evaluation info if necessary.

Parameters
  • allowed_pipelines (list(class)) – A list of PipelineBase subclasses indicating the pipelines allowed in the search. The default of None indicates all pipelines for this problem type are allowed.

  • max_pipelines (int) – The maximum number of pipelines to be evaluated.

  • tuner_class (class) – A subclass of Tuner, to be used to find parameters for each pipeline. The default of None indicates the SKOptTuner will be used.

  • random_state (int, np.random.RandomState) – The random seed/state. Defaults to 0.