pipeline_meta#

Metaclass that overrides creating a new pipeline by wrapping methods with validators and setters.

Module Contents#

Classes Summary#

PipelineBaseMeta

Metaclass that overrides creating a new pipeline by wrapping methods with validators and setters.

Contents#

class evalml.pipelines.pipeline_meta.PipelineBaseMeta[source]#

Metaclass that overrides creating a new pipeline by wrapping methods with validators and setters.

Attributes

FIT_METHODS

[‘fit’, ‘fit_transform’]

METHODS_TO_CHECK

[‘predict’, ‘predict_proba’, ‘transform’, ‘inverse_transform’, ‘get_trend_dataframe’]

PROPERTIES_TO_CHECK

[‘feature_importance’]

Methods

check_for_fit

check_for_fit wraps a method that validates if self._is_fitted is True.

register

Register a virtual subclass of an ABC.

set_fit

Wrapper for the fit method.

classmethod check_for_fit(cls, method)[source]#

check_for_fit wraps a method that validates if self._is_fitted is True.

Parameters

method (callable) – Method to wrap.

Returns

The wrapped method.

Raises

PipelineNotYetFittedError – If pipeline is not yet fitted.

register(cls, subclass)#

Register a virtual subclass of an ABC.

Returns the subclass, to allow usage as a class decorator.

classmethod set_fit(cls, method)#

Wrapper for the fit method.