elphick.sklearn_viz.model_selection.model_selection.plot_model_selection

elphick.sklearn_viz.model_selection.model_selection.plot_model_selection(estimators, datasets, target, pre_processor=None, k_folds=10, title=None)[source]
Parameters:
  • estimators (Union[RegressorMixin, ClassifierMixin, Dict]) – sklearn estimator or a Dict of algorithms to cross-validate, keyed by string name/code.

  • datasets (Union[DataFrame, Dict]) – pandas DataFrame or a dict of DataFrames, keyed by string name/code.

  • target (str) – target column

  • pre_processor (Optional[Pipeline]) – Optional pipeline used to pre-process the datasets.

  • k_folds (int) – The number of cross validation folds.

  • title (Optional[str]) – Optional plot title

Return type:

Figure

Returns:

a plotly GraphObjects.Figure