U 3du4@srddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZd d ZGd d d ZdS) )reduceN) LabelEncoder)check_matplotlib_support)_safe_indexing) is_regressor)check_is_fitted_is_arraylike_not_scalar _num_featurescstd}|r(tjdr(d}t||rVtjdkrV|dkrNd}t|dg}n|dkrjd d dg}n|g}fd d |D}td d|}|dkrtjjdd|d|S)awReturn prediction method from the `response_method` for decision boundary. Parameters ---------- estimator : object Fitted estimator to check. response_method : {'auto', 'predict_proba', 'decision_function', 'predict'} Specifies whether to use :term:`predict_proba`, :term:`decision_function`, :term:`predict` as the target response. If set to 'auto', the response method is tried in the following order: :term:`decision_function`, :term:`predict_proba`, :term:`predict`. Returns ------- prediction_method: callable Prediction method of estimator. classes_rzFMulti-label and multi-output multi-class classifiers are not supported>autopredictzUMulticlass classifiers are only supported when response_method is 'predict' or 'auto'rr Zdecision_functionZ predict_probacsg|]}t|dqSN)getattr).0method estimatorN/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/inspection/_plot/decision_boundary.py 5sz3_check_boundary_response_method..cSs|p|Srr)xyrrr6z1_check_boundary_response_method..Nz' has none of the following attributes: , .) hasattrr r ValueErrorlenr __class____name__join)rresponse_methodZ has_classesmsgZ methods_listZprediction_methodrrr_check_boundary_response_methods(  r&c @sHeZdZdZdddddZdddZed d dd dddd d dZdS)DecisionBoundaryDisplaya Decisions boundary visualization. It is recommended to use :func:`~sklearn.inspection.DecisionBoundaryDisplay.from_estimator` to create a :class:`DecisionBoundaryDisplay`. All parameters are stored as attributes. Read more in the :ref:`User Guide `. .. versionadded:: 1.1 Parameters ---------- xx0 : ndarray of shape (grid_resolution, grid_resolution) First output of :func:`meshgrid `. xx1 : ndarray of shape (grid_resolution, grid_resolution) Second output of :func:`meshgrid `. response : ndarray of shape (grid_resolution, grid_resolution) Values of the response function. xlabel : str, default=None Default label to place on x axis. ylabel : str, default=None Default label to place on y axis. Attributes ---------- surface_ : matplotlib `QuadContourSet` or `QuadMesh` If `plot_method` is 'contour' or 'contourf', `surface_` is a :class:`QuadContourSet `. If `plot_method` is 'pcolormesh', `surface_` is a :class:`QuadMesh `. ax_ : matplotlib Axes Axes with confusion matrix. figure_ : matplotlib Figure Figure containing the confusion matrix. See Also -------- DecisionBoundaryDisplay.from_estimator : Plot decision boundary given an estimator. Examples -------- >>> import matplotlib.pyplot as plt >>> import numpy as np >>> from sklearn.datasets import load_iris >>> from sklearn.inspection import DecisionBoundaryDisplay >>> from sklearn.tree import DecisionTreeClassifier >>> iris = load_iris() >>> feature_1, feature_2 = np.meshgrid( ... np.linspace(iris.data[:, 0].min(), iris.data[:, 0].max()), ... np.linspace(iris.data[:, 1].min(), iris.data[:, 1].max()) ... ) >>> grid = np.vstack([feature_1.ravel(), feature_2.ravel()]).T >>> tree = DecisionTreeClassifier().fit(iris.data[:, :2], iris.target) >>> y_pred = np.reshape(tree.predict(grid), feature_1.shape) >>> display = DecisionBoundaryDisplay( ... xx0=feature_1, xx1=feature_2, response=y_pred ... ) >>> display.plot() <...> >>> display.ax_.scatter( ... iris.data[:, 0], iris.data[:, 1], c=iris.target, edgecolor="black" ... ) <...> >>> plt.show() N)xlabelylabelcCs"||_||_||_||_||_dSrxx0xx1responser(r))selfr+r,r-r(r)rrr__init__s z DecisionBoundaryDisplay.__init__contourfc Kstdddlm}|dkr$td|dkr8|\}}t||}||j|j|jf||_ |dk sj| s|dkrx|j n|}| ||dk s| s|dkr|jn|}||||_|j|_|S)aPlot visualization. Parameters ---------- plot_method : {'contourf', 'contour', 'pcolormesh'}, default='contourf' Plotting method to call when plotting the response. Please refer to the following matplotlib documentation for details: :func:`contourf `, :func:`contour `, :func:`pcolormesh `. ax : Matplotlib axes, default=None Axes object to plot on. If `None`, a new figure and axes is created. xlabel : str, default=None Overwrite the x-axis label. ylabel : str, default=None Overwrite the y-axis label. **kwargs : dict Additional keyword arguments to be passed to the `plot_method`. Returns ------- display: :class:`~sklearn.inspection.DecisionBoundaryDisplay` Object that stores computed values. DecisionBoundaryDisplay.plotrNr0ZcontourZ pcolormeshz:plot_method must be 'contourf', 'contour', or 'pcolormesh')rZmatplotlib.pyplotZpyplotrZsubplotsrr+r,r-Zsurface_Z get_xlabelr(Z set_xlabelZ get_ylabelr)Z set_ylabelZax_figureZfigure_) r. plot_methodaxr(r)kwargsZplt_Z plot_funcrrrplots&     r1dg?r )grid_resolutionepsr4r$r(r)r5cKsXt|jdt||dks0td|d|dksHtd|dd} || krtd| } td | d |dt|} | d krtd | dt|ddd t|ddd }}||||}}||||}}t t |||t |||\}}t |drZ|j gddf}||j dddf<||j dddf<nt j||f}t||}||}|jdkrt |drt}|j|_||}|jdkrt|rtd|dddf}|dkrt |dr|jdnd}|dkr(t |dr$|jdnd}t||||j||d}|jf| |d| S)aPlot decision boundary given an estimator. Read more in the :ref:`User Guide `. Parameters ---------- estimator : object Trained estimator used to plot the decision boundary. X : {array-like, sparse matrix, dataframe} of shape (n_samples, 2) Input data that should be only 2-dimensional. grid_resolution : int, default=100 Number of grid points to use for plotting decision boundary. Higher values will make the plot look nicer but be slower to render. eps : float, default=1.0 Extends the minimum and maximum values of X for evaluating the response function. plot_method : {'contourf', 'contour', 'pcolormesh'}, default='contourf' Plotting method to call when plotting the response. Please refer to the following matplotlib documentation for details: :func:`contourf `, :func:`contour `, :func:`pcolormesh `. response_method : {'auto', 'predict_proba', 'decision_function', 'predict'}, default='auto' Specifies whether to use :term:`predict_proba`, :term:`decision_function`, :term:`predict` as the target response. If set to 'auto', the response method is tried in the following order: :term:`decision_function`, :term:`predict_proba`, :term:`predict`. For multiclass problems, :term:`predict` is selected when `response_method="auto"`. xlabel : str, default=None The label used for the x-axis. If `None`, an attempt is made to extract a label from `X` if it is a dataframe, otherwise an empty string is used. ylabel : str, default=None The label used for the y-axis. If `None`, an attempt is made to extract a label from `X` if it is a dataframe, otherwise an empty string is used. ax : Matplotlib axes, default=None Axes object to plot on. If `None`, a new figure and axes is created. **kwargs : dict Additional keyword arguments to be passed to the `plot_method`. Returns ------- display : :class:`~sklearn.inspection.DecisionBoundaryDisplay` Object that stores the result. See Also -------- DecisionBoundaryDisplay : Decision boundary visualization. ConfusionMatrixDisplay.from_estimator : Plot the confusion matrix given an estimator, the data, and the label. ConfusionMatrixDisplay.from_predictions : Plot the confusion matrix given the true and predicted labels. Examples -------- >>> import matplotlib.pyplot as plt >>> from sklearn.datasets import load_iris >>> from sklearn.linear_model import LogisticRegression >>> from sklearn.inspection import DecisionBoundaryDisplay >>> iris = load_iris() >>> X = iris.data[:, :2] >>> classifier = LogisticRegression().fit(X, iris.target) >>> disp = DecisionBoundaryDisplay.from_estimator( ... classifier, X, response_method="predict", ... xlabel=iris.feature_names[0], ylabel=iris.feature_names[1], ... alpha=0.5, ... ) >>> disp.ax_.scatter(X[:, 0], X[:, 1], c=iris.target, edgecolor="k") <...> >>> plt.show() z.from_estimatorz,grid_resolution must be greater than 1. Got z instead.rz,eps must be greater than or equal to 0. Got r2rzplot_method must be one of z. Got r z#n_features must be equal to 2. Got )ZaxisilocNrr z)Multi-output regressors are not supportedcolumnsr*)r5r4)rr"rrr#r rminmaxnpZmeshgridZlinspacerr=copyZravelZc_r&rr Z transformndimrr>r'Zreshapeshaper8)clsrXr:r;r4r$r(r)r5r6Zpossible_plot_methodsZavailable_methodsZ num_featuresZx0x1Zx0_minZx0_maxZx1_minZx1_maxr+r,ZX_gridZ pred_funcr-encoderZdisplayrrrfrom_estimatorsne              z&DecisionBoundaryDisplay.from_estimator)r0NNN)r" __module__ __qualname____doc__r/r8 classmethodrJrrrrr'@sI 7r') functoolsrZnumpyrBZ preprocessingrutilsrrbaserZutils.validationrr r r&r'rrrrs     0