_nd<,ddlZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z ej dZejd d d ied fd d iedfgdZdZdZejdddgdZejd ddZdZdZdZdS)N) load_iris)DecisionTreeClassifier)shuffle)assert_allcloseassert_array_equal)learning_curve)LearningCurveDisplayc6ttdddiS)NT) return_X_y random_stater)rrGlib/python3.11/site-packages/sklearn/model_selection/tests/test_plot.pydatar s I... ?Q ? ??rzparams, err_type, err_msgstd_display_styleinvalidzUnknown std_display_style: score_typezUnknown score_type:c|\}}td}gd}tj||5tj|||fd|i|ddddS#1swxYwYdS)zCCheck that we raise a proper error when passing invalid parameters.rr g333333?g333333?g?)match train_sizesN)rpytestraisesr from_estimator) pyplotrparamserr_typeerr_msgXy estimatorrs r1test_learning_curve_display_parameters_validationr#s DAq&A666I!//K xw / / /  + q!  )4 8>                     sAAAcJ|\}}td}gd}tj||||}ddl}|jJt |jtsJ|jD]}t ||jj sJt |j tsJ|j D]8} t | |j j sJ| dksJ9|jdksJ|jdksJ|jdksJ|j\} } | d gksJt'||||\} } }t)|j| t-|j| t-|j|dS) z:Check the default usage of the LearningCurveDisplay class.rrrrNg?Scorez%Number of samples in the training setTesting metric)rr r matplotlib errorbar_ isinstancelines_listlinesLine2D fill_between_ collectionsPolyCollection get_alpha score_nameax_ get_xlabel get_ylabelget_legend_handles_labelsrrrr train_scores test_scores)rrr r!r"rdisplaympllinefill_ legend_labelstrain_sizes_absr8r9s r)test_learning_curve_display_default_usagerA&s DAq&A666I!//K"11a[G   $ $ $ gnd + ++++22$ 0111111 g+T 2 2222%''$ >?????~~3&&&&&   ( ( ( ( ; ! ! # #'N N N N N ; ! ! # #w . . . .{<<>>A} -. . . . .1?1a[222.O\;w*O<<<G(,777G'55555rc|\}}tdd}gd}d}tj|||||}|jdd}|dksJ|jdksJd}tj|||||}|jdd} | dksJt| | |jdksJd}tj|||||}|jdksJ| | |jdksJ|jdddksJd S) zaCheck the behaviour of the `negate_score` parameter calling `from_estimator` and `plot`. r max_depthr rF)r negate_scorer&T)rFN) rr rr+get_dataallr4r6rplot) rrr r!r"rrFr:positive_scoresnegative_scoress r(test_learning_curve_display_negate_scorerLMs DAq&CCCI!//KL"1  ! GnQ'002215O q % % ' '''' ; ! ! # #w . . . .L"11a[|GnQ'002215O q % % ' ''''Oo%5666 ; ! ! # #w . . . .L"1  ! G ; ! ! # #w . . . . LL,.L/// ; ! ! # #w . . . . N1  & & ( ( +a / 4 4 6 666666rzscore_name, ylabel)Nr&)AccuracyrMc(|\}}td}gd}tj|||||}|j|ksJ|\}}tdd}gd}tj|||||}|j|ksJdS)zGCheck that we can overwrite the default score name shown on the y-axis.rrr)rr3rCrDN)rr rr4r6r3) rrr3ylabelr r!r"rr:s r&test_learning_curve_display_score_namerPzs DAq&A666I!//K"11a[ZG ; ! ! # #v - - - - DAq&CCCI!//K"11a[ZG   ' ' ' ' ' 'r)Nerrorbarc|\}}td}gd}t||||\}}} d} tj||||| |} | j\} } | dgksJ|Ft | jd ksJ| jJ| jd \}}nP| jJt | jd ksJ| jdj d \}}t||t|| d d } tj||||| |} | j\} } | d gksJ|Ft | jd ksJ| jJ| jd \}}nP| jJt | jd ksJ| jdj d \}}t||t|| d d } tj||||| |} | j\} } | dd gksJ|ht | jdksJ| jJ| jd \}}| jd  \}}n}| jJt | jdksJ| jdj d \}}| jd j d \}}t||t|| d t||t|| d dS)z:Check the behaviour of setting the `score_type` parameter.rrrr%train)rrrzTraining metricNrC)axistestr'both)rrr rr4r7lenr+r)rGr-rrmean)rrrr r!r"rr@r8r9rr:r> legend_labelx_datay_data x_data_train y_data_train x_data_test y_data_tests r&test_learning_curve_display_score_typerasI DAq&A666I!//K1?1a[222.O\;J"1  + Gk;;==OA| -. . . . . 7>""a'''' ((( *3355~%%%7$%%**** *1-3A6??AAv///FL--1-55666J"1  + Gk;;==OA| ,- - - - - 7>""a'''' ((( *3355~%%%7$%%**** *1-3A6??AAv///FK,,!,44555J"1  + Gk;;==OA| -/?@ @ @ @ @ 7>""a'''' (((%,^A%6%?%?%A%A" l#*>!#4#=#=#?#? [[~%%%7$%%****%,%6q%9%?%B%K%K%M%M" l#*#4Q#7#=a#@#I#I#K#K [|_555L,"3"3"3";";<<<{O444K!1!1q!1!9!9:::::rc|\}}td}gd}tj||||d}|jdksJ|jdksJtj||||d}|jdksJ|jdksJd S) z1Check the behaviour of the parameter `log_scale`.rrrT)r log_scaleloglinearFN)rr rr4 get_xscale get_yscale)rrr r!r"rr:s r%test_learning_curve_display_log_scalerhs DAq&A666I!//K"11a[DG ; ! ! # #u , , , , ; ! ! # #x / / / /"11a[EG ; ! ! # #x / / / / ; ! ! # #x / / / / / /rcj|\}}td}ddl}gd}d}tj|||||}t |jdksJt |jd|jjsJ|j J|j J|j \} } t | dksJd}tj|||||}t |jdksJt |jd|jjsJ|j Jt |j dksJt |j d|j jsJ|j \} } t | dksJd}tj|||||}|jJt |j dksJt |j d|jjsJ|j J|j \} } t | dksJdS) z9Check the behaviour of the parameter `std_display_style`.rrNr)rrrC fill_betweenrQ)rr(r rrXr+r*r-r.r)r/r4r7r0r1 containerErrorbarContainer) rrr r!r"r;rrr:r>rZs r-test_learning_curve_display_std_display_stylerms DAq&A666I!//K"1  + G w~  ! # # # # gnQ')9 : ::::   $ $ $  ( ( (k;;==OA| |   ! ! ! !&"1  + G w~  ! # # # # gnQ')9 : ::::   $ $ $ w$ % % * * * * g+A.0N O OOOOk;;==OA| |   ! ! ! !""1  + G > ! ! ! w ! !Q & & & & g'*CM,K L LLLL  ( ( (k;;==OA| |   ! ! ! ! ! !rc |\}}td}gd}d}ddi}ddd}tj||||||| } | jddksJt | jdgd gd }ddi} tj|||||| } | jdj ddksJd S)zuCheck the behaviour of the different plotting keyword arguments: `line_kw`, `fill_between_kw`, and `errorbar_kw`.rrrrjcolorred?)roalpha)rrline_kwfill_between_kw)rqrurqrQ)rr errorbar_kwN) rr rr+ get_colorrr/ get_facecolorr)r-) rrr r!r"rrrsrtr:rvs r'test_learning_curve_display_plot_kwargsry=s: DAq&A666I!//K&G %44O"1  +'G >!  & & ( (E 1 1 1 1a ..00    #E"K"1  + G  Q  %a ( 2 2 4 4 = = = = = =r)rsklearn.datasetsr sklearn.treer sklearn.utilsrsklearn.utils._testingrrsklearn.model_selectionrr fixturermark parametrize ValueErrorr#rArLrPrarhrmryr rrrs &&&&&&//////!!!!!!FFFFFFFF222222888888@@@ y ):7ST  "J0EF     $6$6$6N*7*7*7Z?,DE(((,,.@AAW;W;BAW;t000*7"7"7"t%>%>%>%>%>r