Source code for evalml.pipelines.regression.baseline_regression

from evalml.pipelines import RegressionPipeline


[docs]class BaselineRegressionPipeline(RegressionPipeline):
    """Baseline Pipeline for regression problems"""
    _name = "Baseline Regression Pipeline"
    component_graph = ["Baseline Regressor"]


[docs]class MeanBaselineRegressionPipeline(RegressionPipeline):
    """Baseline Pipeline for regression problems"""
    _name = "Mean Baseline Regression Pipeline"
    component_graph = ["Baseline Regressor"]
    custom_hyperparameters = {"strategy": ["mean"]}