/
baseline_multiclass.py
27 lines (20 loc) · 1.1 KB
/
baseline_multiclass.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from evalml.pipelines import MulticlassClassificationPipeline
class BaselineMulticlassPipeline(MulticlassClassificationPipeline):
"""Baseline Pipeline for multiclass classification."""
custom_name = "Baseline Multiclass Classification Pipeline"
component_graph = ["Baseline Classifier"]
custom_hyperparameters = None
def __init__(self, parameters, random_seed=0):
super().__init__(self.component_graph,
custom_name=self.custom_name,
parameters=parameters,
custom_hyperparameters=self.custom_hyperparameters,
random_seed=random_seed)
def new(self, parameters, random_seed=0):
return self.__class__(parameters, random_seed=random_seed)
def clone(self):
return self.__class__(self.parameters, random_seed=self.random_seed)
class ModeBaselineMulticlassPipeline(BaselineMulticlassPipeline):
"""Mode Baseline Pipeline for multiclass classification."""
custom_name = "Mode Baseline Multiclass Classification Pipeline"
custom_hyperparameters = {"strategy": ["mode"]}