component_base_meta#

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

Module Contents#

Classes Summary#

ComponentBaseMeta

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

Contents#

class evalml.pipelines.components.component_base_meta.ComponentBaseMeta[source]#

Metaclass that overrides creating a new component 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

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.

It raises an exception if False and calls and returns the wrapped method if True.

Parameters

method (callable) – Method to wrap.

Returns

The wrapped method.

Raises

ComponentNotYetFittedError – If component 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.