evalml.automl.AutoClassificationSearch

Inheritance diagram of AutoClassificationSearch
class evalml.automl.AutoClassificationSearch(objective=None, multiclass=False, max_pipelines=None, max_time=None, patience=None, tolerance=None, allowed_model_families=None, cv=None, tuner=None, detect_label_leakage=True, start_iteration_callback=None, add_result_callback=None, additional_objectives=None, random_state=0, n_jobs=-1, verbose=True, optimize_thresholds=False)[source]

Automatic pipeline search class for classification problems

Methods

__init__

Automated classifier pipeline search

describe_pipeline

Describe a pipeline

get_pipeline

Retrieves trained pipeline

search

Find best classifier

Attributes

best_pipeline

Returns the best model found

full_rankings

Returns a pandas.DataFrame with scoring results from all pipelines searched

rankings

Returns a pandas.DataFrame with scoring results from the highest-scoring set of parameters used with each pipeline.