evalml.automl.AutoClassificationSearch

Inheritance diagram of AutoClassificationSearch
class evalml.automl.AutoClassificationSearch(objective=None, max_pipelines=None, max_time=None, patience=None, tolerance=None, cv=None, allowed_pipelines=None, allowed_model_families=None, start_iteration_callback=None, add_result_callback=None, additional_objectives=None, random_state=0, n_jobs=-1, tuner_class=None, verbose=True, optimize_thresholds=False, multiclass=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

data_check_results

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.