B 0dV @sddlZddlmZddlmZddlZddlZddlm Z ddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddl"m#Z#ddZ$ddZ%ej&'dge(dfd d!d"ge(dfe)ge(dfe)d d!d"ge(dfej**d#e(dfgd$d%Z+ej&'d&id#d#d'gd(d)Z,d*d+Z-d,d-Z.d.d/Z/ej&'d0d1did2fd3did4fgd5d6Z0d7d8Z1d9d:Z2d;d<Z3d=d>Z4d?d@Z5dAdBZ6ej&'dCdDd#dge)dDd#dge7dDd#dggdEdFZ8dGdHZ9dIdJZ:dKdLZ;dMdNZdSdTZ?dUdVZ@dWdXZAdYdZZBd[d\ZCd]d^ZDd_d`ZEdadbZFdcddZGdedfZHdgdhZIdS)iN) defaultdict)partial)assert_array_equal)assert_almost_equal)assert_array_almost_equal)make_classification)make_multilabel_classification)make_hastie_10_2)make_regression) make_blobs)make_friedman1)make_friedman2)make_friedman3)make_low_rank_matrix) make_moons) make_circles)make_sparse_coded_signal)make_sparse_uncorrelated)make_spd_matrix)make_swiss_roll) make_s_curve)make_biclusters)make_checkerboard)assert_all_finitecCsRddg}tdddddddddd|d d \}}|ddgksmaker!r)r%r$ n_clustersrr&r9r:ZsignsZ unique_signsZ cluster_indexZclusters_by_classZclusterclsZclustersZcentroidr;r;r<-test_make_classification_informative_featuresSs            rYzweights, err_type, err_msgz:Weights specified but incompatible with number of classes.g?g?g?rc Cs(tj||dt|dWdQRXdS)N)match)r))rSrTr)r)Zerr_typerCr;r;r<%test_make_classification_weights_typesr[kwargs)r$r!cCsVtfddgdd|\}}tftddgdd|\}}t||t||dS)Ng?g?r)r)r*)rr4arrayr)r\X1y1X2y2r;r;r<1test_make_classification_weights_array_or_list_oks" rbc CsxtddD]\}}tddddd|d\}}|jd ks>td |s\td d |Dd ks\ttdd |D|ksvttdd |Ddks tq WdS)N)TF)rrrrrrF)rr r$r*return_indicatorallow_unlabeled)rrzX shape mismatchcSsg|] }t|qSr;)max).0r:r;r;r< szHtest_make_multilabel_classification_return_sequences..r-cSsg|] }t|qSr;)rG)rfr:r;r;r<rgscSsg|] }t|qSr;)rG)rfr:r;r;r<rgs)rLrr3r2remin)rd min_lengthr9Yr;r;r<4test_make_multilabel_classification_return_sequencess rkcCsxltddD]^\}}tdddd|d\}}|jdkstd |jd ksPtd t|s tq WdS) N)TF)rrr,rrrsparse)rr r$r*rcrd)r,rzX shape mismatch)r,rzY shape mismatch)rLrr3r2spissparse)rdrir9rjr;r;r<;test_make_multilabel_classification_return_indicator_sparse s rqzparams, err_msgr$z 'n_classes' should be an integerlengthz'length' should be an integerc Cs(tjt|dtf|WdQRXdS)N)rZ)rSrTrUr)paramsrCr;r;r<3test_make_multilabel_classification_valid_argumentssrtcCsPtddd\}}|jdks"td|jdks4tdt|jdksLtd dS) Nrr)rr*)rr+zX shape mismatch)rzy shape mismatch)r-zUnexpected number of classes)r r3r2r4r5)r9r:r;r;r<test_make_hastie_10_2(sruc Cstddddddddd \}}}|jd ks0td |jd ksBtd |jdksTtdt|dkdksltdtt|t||dddtddd\}}|jdkstdS)Nrr+rrTgg?r)rr r!effective_rankcoefZbiasnoiser*)rr+zX shape mismatch)rzy shape mismatch)r+zcoef shape mismatchz)Unexpected number of informative featuresr)rE)rr )rr)r r3r2r6rr4stddot)r9r:cr;r;r<test_make_regression/s  r|c Cstdddddddd\}}}|jdks.td |jd ks@td |jd ksRtd tt|dkddtt|t||ddddS)Nrr+rTg?r)rr r! n_targetsrwrxr*)rr+zX shape mismatch)rrzy shape mismatch)r+rzcoef shape mismatchgz)Unexpected number of informative featuresr)rE) r r3r2rr6rr4ryrz)r9r:r{r;r;r< test_make_regression_multitargetHsr~cCstdddg}tddgddgddgg}tddd||d \}}|jd ksTtd |jd ksftd t|jdks~tdx\}}|dkr4ddgnddg}||d}t|dd d qWdS) NrF)r?rgg?g?r-z$Point is not on expected unit circle)rC)rrLr6r)r9r:xlabelcenterdist_sqrr;r;r<test_make_moonsOs rc Cstdd\}}t|dkdkr2t|dkdks:td|jdksLtd |jd ks^td tjtd d tdddgdWdQRXtjtd d tddWdQRXdS)N)r)rrrrrz$Number of samples in a moon is wrong) r-zX shape mismatch)rzy shape mismatchz8`n_samples` can be either an int or a two-element tuple.)rZr-r)r+)rr4r6r2r3rSrTrU)r9r:r;r;r<test_make_moons_unbalancedYs&  rc Cs<d}xdD]\}}}t|dd|d\}}|j|dfks>td|j|fksRtdddg}x\t||D]N\}}||d} |d krd n|d} |d krd n|d} t| | d d qfW||d kj|dfkstd ||dkj|dfks td q WtttddWdQRXtttd dWdQRXdS)Ng333333?))rrr@)rr@r@F)r?rxfactorr-zX shape mismatchzy shape mismatchgrg?zPoint is not on expected circle)rCz1Samples not correctly distributed across circles.rg{Gz)r) rr3r2rLr6rrSrTrU) rrZn_outerZn_innerr9r:rrrrZdist_expr;r;r<test_make_circlesns0       rc Cstdd\}}t|dkdks(tdt|dkdksBtd|jd ksTtd |jd ksftd tjtd dtdddgdWdQRXtjtd dtd dWdQRXdS)N)r-r)rrr-z*Number of samples in inner circle is wrongrrz*Number of samples in outer circle is wrong)r+r-zX shape mismatch)r+zy shape mismatchz8`n_samples` can be either an int or a two-element tuple.)rZr)rr4r6r2r3rSrTrU)r9r:r;r;r<test_make_circles_unbalanceds  r)Jr collectionsr functoolsrnumpyr4rSZ scipy.sparsernroZsklearn.utils._testingrrrZsklearn.datasetsrrr r r r r rrrrrrrrrrrZsklearn.utils.validationrr=rYmarkZ parametrizerUr]randomr[rbrkrmrqrtrur|r~rrrtuplerrrrrrrrrrrrrrrrrrr;r;r;r<s                         0c    2        !