U _{jvF@s<ddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZmZmZddddgdd d d gdd dd gddddggZejddddddgddddgddddgddddggdfdddddgddddgddddgddddggdfdddddgddddgddddgddddggdfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfgejdddZddZejddgddZddZejddddddgddddgddddgddddggdfdddddgddddgddddgddddggdfdddddgddddgddddgddddggdfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfdddddgddddgddddgddddggddddgfgejdd d!Zejd"d#d$Zejdejdddgd%d&Zejddddgd'd(Zd)d*Zejd+edd,d-d.Zd/d0Z ejd1dddddddgddddddgddddd2d2gfdddddddgddddddgdddddd2gfdddddddgddddddgddddd2d2gfgejdd3d4Z!ejd5dd6d7d8d gd d9d:d gd d;d6d gd d;d6dggfddd?gdd?gd@dAdBdCgdDdEdFdGggfdd6d7d8dHgd d9d:dIgd d;d6dJgd d;d6dJggfgejdejdKdLdMdNgdOdPZ"ejdejddddgdQdRZ#dSdTZ$ejdUddddgfddddgfgdVdWZ%dXdYZ&ejdZej'ej(ej)gejd[dej(ej)gejdKdLdMdNgd\d]Z*ejd^ej'ej(ej)gejdKdLdMdNgd_d`Z+dadbZ,ejdcdMdddeedDfdNdfdeedDfdLdgdeedDfgdhdiZ-ejddddgdjdkZ.ejdddgdldmZ/dS)nN)clone)KBinsDiscretizer OneHotEncoder)assert_allcloseassert_allclose_dense_sparseassert_array_almost_equalassert_array_equal?@gg @?g@z!strategy, expected, sample_weightuniformZkmeansquantilez0ignore:In version 1.5 onwards, subsample=200_000cCs0tdd|d}|jt|dt||tdS)Nrordinaln_binsencodestrategy sample_weight)rfitXr transform)rexpectedrestr!n/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sklearn/preprocessing/tests/test_discretization.pytest_fit_transforms$r#cCsRtddtttdgddttddtjjttksNt dS)Nrrr) r fit_transformrnparrayrn_bins_dtypeintAssertionErrorr!r!r!r"test_valid_n_bins<sr,rc CsLtjttd}td|d}d}tjt|d|jt|dW5QRXdS)z=Check that we raise an error when the wrong strategy is used.)shaperrrzK`sample_weight` was provided but it cannot be used with strategy='uniform'.matchrN) r&Zoneslenrrpytestraises ValueErrorr)rrr err_msgr!r!r"1test_kbinsdiscretizer_wrong_strategy_with_weightsBs  r6c Cstdd}t|d}d}tjt|d|tW5QRXdddg}t|d}d}tjt|d|tW5QRXddddg}t|d}d}tjt|d|tW5QRXd dd dg}t|d}d }tjt|d|tW5QRXdS) N)r@r$z:n_bins must be a scalar or array of shape \(n_features,\).r/rrz{KBinsDiscretizer received an invalid number of bins at indices 0, 3. Number of bins must be at least 2, and must be an int.g@z{KBinsDiscretizer received an invalid number of bins at indices 0, 2. Number of bins must be at least 2, and must be an int.)r&fullrr2r3r4r%r)rr r5r!r!r"test_invalid_n_bins_arrayNs,        r:cCstddddgd|djt|d}t||tttjd}|jj|fksRt t |j|j D]\}}|j|dfks`t q`dS)Nrrrrrr) rrrrrr&r'r- bin_edges_r+zipr()rrrr Z n_features bin_edgesrr!r!r"test_fit_transform_n_bins_arrayts' r>z&ignore: Bins whose width are too smallcCstdgdgdgdgdgdgg}tddd d }|j|ddddd d gd t|jd ddddgt||d gdgdgdgdgdggdS)z;Check the impact of `sample_weight` one computed quantiles.r r rrii rrrrrg?r8N)r&r'rrrr;rrr r!r!r"*test_kbinsdiscretizer_effect_sample_weights "rBcCsLtdd|d}tjddddgtjd}t|}|jt|dt||dS) z7Make sure that `sample_weight` is not changed in place.rrrrrr)rN)rr&r'float64copyrrr)rr rZsample_weight_copyr!r!r"/test_kbinsdiscretizer_no_mutating_sample_weights  rFc Cstdtddgddgddgddgg}t|ddd}d }tjt|d ||W5QRX|j ddkstt | |}t |dddft |jddS) Nalwaysrr r rrr)rrrz2Feature 0 is constant and will be replaced with 0.r/)warnings simplefilterr&r'rr2warns UserWarningrr(r+rrZzerosr-)rrr warning_messageXtr!r!r"test_same_min_maxs " rNc Csttd}tdd}tt||W5QRXtdd}||ddtt||W5QRXdS)Nr7rr$r r) r&arangerr2r3r4rreshaperrAr!r!r"test_transform_1d_behaviors     rQi cCsdtdddddgdd}tdddddgdd}|d |}td d d |}t||dS) Nr8@g@g @g$@r rrr?rrrr)r&r'rPrr%r)rRZX_initZ Xt_expectedrrMr!r!r"test_numeric_stabilitys  rVcCstddddgddt}|t}tddddgddt}|t}t|rVtttdddDd d  ||tddddgd dt}|t}t|stttd ddDd d  | | dS)NrrrrU onehot-densecSsg|]}t|qSr!r&rO.0rRr!r!r" sz'test_encode_options..)rrrrF) categoriesZ sparse_outputonehotcSsg|]}t|qSr!rXrYr!r!r"r[sT) rrrrspissparser+rrr%Ztoarray)r ZXt_1ZXt_2ZXt_3r!r!r"test_encode_optionss4     r`z8strategy, expected_2bins, expected_3bins, expected_5binsr7cCstddddddgdd}td|d d }||}t||td|d d }||}t||td |d d }||}t||dS) NrrrrrSr?r rrrrr)r&r'rPrr%rZravel)rZexpected_2binsZexpected_3binsZexpected_5binsrr rMr!r!r"test_nonuniform_strategiess    rczstrategy, expected_invgr8g g@grTgg@g ggg @gg?g?g@gg?gr@g?rrr]rWcCs0td||d}|t}||}t||dS)Nrra)rr%rinverse_transformr)rrZ expected_invkbdrMXinvr!r!r"test_inverse_transforms$  rgcCstddddgdddf}td|dd}||tdd gdddf}||}t|jdd d|jt|jdd dgdS) Nrrrrr7rrar rb)Zaxis) r&r'rrrrmaxr(min)rrreZX2ZX2tr!r!r" test_transform_outside_fit_rangeGs  rjcCstddddgdddf}|}tddd}||}t|||}||}t||t|tdgdgd gd ggdS) NrrrrrrUrr r )r&r'rErr%rrd)rZX_beforer rMZ Xt_beforerfr!r!r"test_overwriteTs     rkzstrategy, expected_bin_edgesc Csddgdgdgdgdgdgg}td|d}d}tjt|d||W5QRXt|jd|dS)Nrrr.'Consider decreasing the number of bins.r/)rr2rJrKrrr;)rZexpected_bin_edgesrrerLr!r!r"test_redundant_binsbs  rmc Cstdddgdd}tddddddg}td d d gdd}td d d d}d}tjt|d||W5QRXt|j d |t| ||dS)Ng?gffffff?r rgq= ףp?g= ףp=?gzG?gp= ף?rr7r?rrrrlr/) r&r'rPrr2rJrKrrr;r)rr=rMrerLr!r!r"!test_percentile_numeric_stabilitynsrnin_dtype out_dtypecCsrtjt|d}td||d}|||dk r4|}n"|dkrP|jtjkrPtj}n|j}||}|j|ksnt dSNrCr)rrr)) r&r'rrrr)float16rDrr+)rorprX_inputreZexpected_dtyperMr!r!r"test_consistent_dtype{s  rt input_dtypecCsdtjt|d}td|tjd}||||}td|tjd}||||}t||dSrq) r&r'rrfloat32rrrDr)rurrsZkbd_32ZXt_32Zkbd_64ZXt_64r!r!r"test_32_equal_64s    rwcCstddddgdd}tdddd }||t|}|jdd ||t|jd |jd D]\}}tj ||qf|jj |jj kst dS) Nr r r r rr?rrr subsampler) r&r'rPrrr set_paramsr<r;Ztestingrr-r+)rZ kbd_defaultZkbd_without_subsamplingZbin_kbd_defaultZbin_kbd_with_subsamplingr!r!r"'test_kbinsdiscretizer_subsample_defaults    r{zencode, expected_namescCs.g|]&}tdD]}d|dt|qqSr7feat_rangefloatrZcol_idZbin_idr!r!r"r[s r[cCs.g|]&}tdD]}d|dt|qqSr|rrr!r!r"r[s cCsg|]}d|qSr}r!)rZrr!r!r"r[scCsdddgdddgdddgdd dgg}td |d |}||}d d tdD}||}|jd|jdksttt||d S)z[Check get_feature_names_out for different settings. Non-regression test for #22731 r rr r rrrrr7rUcSsg|]}d|qSrr!rYr!r!r"r[sz>test_kbinsdiscrtizer_get_feature_names_out..N)rrrrZget_feature_names_outr-r+r)rZexpected_namesrrerMZinput_featuresZ output_namesr!r!r"*test_kbinsdiscrtizer_get_feature_names_outs$  rcCsjtj|dd}t|d|d}||t|}|jdd||t|j d|j ddddS) N)irriP)rry random_staterxrg{Gz?)Zrtol) r&random RandomState random_samplerrrrzrr;)rZglobal_random_seedrZkbd_subsamplingZkbd_no_subsamplingr!r!r"test_kbinsdiscretizer_subsamples   rc CsFtjdd}t|dd}tjtdd||W5QRXdS)Nr)dr)rrz)subsample=200_000 will be used by defaultr/) r&rrrrr2rJ FutureWarningr)rrrer!r!r"test_kbd_subsample_warnings r)0rHnumpyr&r2Z scipy.sparsesparser^ZsklearnrZsklearn.preprocessingrrZsklearn.utils._testingrrrrrmarkZ parametrizefilterwarningsr#r,r6r:r>rBrFrNrQrrVr`rcrgrjrkrmrnrrrvrDrtrwr{rrrr!r!r!r"s&  ,000* * * * * "   &000* * * %       ...                !