evalml.pipelines.components.BaselineRegressor

class evalml.pipelines.components.BaselineRegressor(strategy='mean', random_state=0, **kwargs)

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 = {}

default_parameters = {'strategy': 'mean'}

Instance attributes

feature_importance	Return an attribute of instance, which is of type owner.
parameters	Returns the parameters which were used to initialize the component

Methods:

__init__	Baseline regressor that uses a simple strategy to make predictions.
clone	Constructs a new component with the same parameters
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.

evalml.pipelines.components.XGBoostRegressor.predict_proba evalml.pipelines.components.BaselineRegressor.__init__