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

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)

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.