pipeline_meta

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’]

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

classmethod check_for_fit(cls, method)[source]

check_for_fit wraps a method that validates if self._is_fitted is True. It raises an exception if False and calls and returns the wrapped method if True.

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)