Source code for evalml.pipelines.classification.baseline_multiclass

from evalml.pipelines import MulticlassClassificationPipeline


class BaselineMulticlassPipeline(MulticlassClassificationPipeline):
    """"Baseline Pipeline for multiclass classification"""
    custom_name = "Baseline Multiclass Classification Pipeline"
    component_graph = ["Baseline Classifier"]


class ModeBaselineMulticlassPipeline(MulticlassClassificationPipeline):
    """"Mode Baseline Pipeline for multiclass classification"""
    custom_name = "Mode Baseline Multiclass Classification Pipeline"
    component_graph = ["Baseline Classifier"]
    custom_hyperparameters = {"strategy": ["mode"]}