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 columnpre_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