evalml.automl.AutoSearchBase

class evalml.automl.AutoSearchBase(problem_type=None, 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=None)

Base class for AutoML searches.

Methods

__init__	Initialize self.
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.