evalml.automl.AutoMLSearch.search

AutoMLSearch.search(X, y, data_checks='auto', feature_types=None, raise_errors=True, show_iteration_plot=True)

Find best classifier

Parameters

• X (pd.DataFrame) – the input training data of shape [n_samples, n_features]

• y (pd.Series) – the target training labels of length [n_samples]

• feature_types (list, optional) – list of feature types, either numerical or categorical. Categorical features will automatically be encoded

• raise_errors (boolean) – If True, raise errors and exit search if a pipeline errors during fitting. If False, set scores for the errored pipeline to NaN and continue search. Defaults to True.

• show_iteration_plot (boolean, True) – Shows an iteration vs. score plot in Jupyter notebook. Disabled by default in non-Jupyter enviroments.

• data_checks (DataChecks, list(Datacheck), str, None) – A collection of data checks to run before automl search. If data checks produce any errors, an exception will be thrown before the search begins. If “disabled” or None, no data checks will be done. If set to “auto”, DefaultDataChecks will be done. Default value is set to “auto”.

Returns

self