Learner

class falcon.abstract.Learner(task: str, **kwargs: Any)

Subclass of PipelineElement. Learners are task aware pipeline elements that act as wrappers around models and responsible for tuning of the hyperparameters.

__init__(task: str, **kwargs: Any) → None

Parameters

task (str) – current ML task

abstract fit(X: ndarray[Any, dtype[ScalarType]], y: ndarray[Any, dtype[ScalarType]], *args: Any, **kwargs: Any) → Any

Parameters

• X (npt.NDArray) – features

• y (npt.NDArray) – targets

Returns

usually None

Return type

Any

fit_pipe(X: Any, y: Any, *args: Any, **kwargs: Any) → Any

Equivalent of fit method that is used for elements chaining inisde pipeline during training.

Parameters

• X (Any) – features

• y (Any) – targets

Returns

usually None

Return type

Any

forward(X: Any, *args: Any, **kwargs: Any) → Any

Equivalent of predict method that is used for elements chaining inside pipeline during inference.

Parameters

X (Any) – featrues

Returns

predictions

Return type

Any

abstract get_input_type() → Type

Returns

Input types

Return type

Type

abstract get_output_type() → Type

Returns

Output types

Return type

Type

abstract predict(X: ndarray[Any, dtype[ScalarType]], *args: Any, **kwargs: Any) → ndarray[Any, dtype[ScalarType]]

Parameters

X (npt.NDArray) – features

Returns

predictions

Return type

npt.NDArray