evalml.pipelines.components.BaselineClassifier

class evalml.pipelines.components.BaselineClassifier(strategy='mode', random_state=0, **kwargs)

Classifier that predicts using the specified strategy.

This is useful as a simple baseline classifier to compare with other classifiers.

name = 'Baseline Classifier'

model_family = 'baseline'

supported_problem_types = [<ProblemTypes.BINARY: 'binary'>, <ProblemTypes.MULTICLASS: 'multiclass'>]

hyperparameter_ranges = {}

default_parameters = {'strategy': 'mode'}

Instance attributes

classes_	Returns class labels.
feature_importance	Returns importance associated with each feature.
needs_fitting
parameters	Returns the parameters which were used to initialize the component

Methods:

__init__	Baseline classifier 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
load	Loads component at file path
predict	Make predictions using selected features.
predict_proba	Make probability estimates for labels.
save	Saves component at file path

Class Inheritance

evalml.pipelines.components.XGBoostClassifier.save evalml.pipelines.components.BaselineClassifier.__init__