streamline.models.naive_bayes module

class streamline.models.naive_bayes.NaiveBayesClassifier(cv_folds=3, scoring_metric='balanced_accuracy', metric_direction='maximize', random_state=None, cv=None, n_jobs=None)

Bases: BaseModel, ABC

Base Model Class for all ML Models

Parameters:

• model –

• model_name –

• cv_folds –

• scoring_metric –

• metric_direction –

• random_state –

• cv –

• sampler –

• n_jobs –

color = 'silver'

model_name = 'Naive Bayes'

objective(trial, params=None)

Unimplemented objective function stub, needs to be overridden :param trial: optuna trial object :param params: dict of optional params or None

small_name = 'NB'