evalml.pipelines.components.BaselineRegressor

Inheritance diagram of BaselineRegressor
class evalml.pipelines.components.BaselineRegressor(strategy='mean', random_state=0)[source]

Regressor that predicts using the specified strategy.

This is useful as a simple baseline regressor to compare with other regressors.

name = 'Baseline Regressor'
model_family = 'baseline'
supported_problem_types = [<ProblemTypes.REGRESSION: 'regression'>]
hyperparameter_ranges = {}

Instance attributes

feature_importances

Returns feature importances.

Methods:

__init__

Baseline regressor that uses a simple strategy to make predictions.

describe

Describe a component and its parameters

fit

Fits component to data

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.