evalml.pipelines.components.BaselineClassifier

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

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

Inheritance diagram of BaselineClassifier