a j~@s<ddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZddlmZddlmZddlmZddlmZdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'dd l(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4GdddeZ5GdddeZ6GdddeZ7GdddeZ8GdddeZ9Gddde8Z:Gddde8e9Z;Gd d!d!e;ZGd&d'd'Z?Gd(d)d)eZ@d*d+ZAd,d-ZBd.d/ZCd0d1ZDd2d3ZEd4d5ZFd6d7ZGd8d9ZHd:d;ZIdd?ZKd@dAZLdBdCZMejNOdDe&dEfe e&dFdGdHgidEfe"dIe&fgdEfe"dJe e&dFdGdHgifgdEfe'dKfe e'dFdGdHgidKfe"dLe'fgdKfe"dMe e'dFdGdHgifgdKfgdNdOZPejNOdDe'dEfe e'dFdGdHgidEfe"dLe'fgdEfe"dMe e'dFdGdHgifgdEfe&dKfe e&dFdGdHgidKfe"dIe&fgdKfe"dJe e&dFdGdHgifgdKfgdPdQZQejNOdDedEfe edRdSdTgidEfe"dUefgdEfe"dVe edRdSdTgifgdEfe&dKfe e&dFdGdHgidKfe"dIe&fgdKfe"dJe e&dFdGdHgifgdKfgdWdXZRdYdZZSd[d\ZTd]d^ZUejNOd_e)d`ddae jVddbfe*d`ddae jWddbfgdcddZXdedfZYdgdhZZdidjZ[Gdkdldle)Z\dmZ]dndoZ^Gdpdqdqe)Z_drdsZ`dtduZaGdvdwdwZbGdxdydyebeZcdzd{Zdd|d}ZeGd~ddeZfddZgddZhddZiddZjddZkddZlddZmddZnddZoddZpGdddZqGdddeqeZrejNOdeergddZsddZtejNOdgdddZue dEdddZve dEdddZwdS)N)assert_allclose)config_contextdatasets) BaseEstimator OutlierMixinTransformerMixinclone is_classifier is_clustereris_outlier_detector is_regressor)KMeans)PCA)IsolationForest)InconsistentVersionWarning) GridSearchCV)Pipeline)StandardScaler)SVCSVR)DecisionTreeClassifierDecisionTreeRegressor) MockDataFrame)_get_output_config)_convert_containerassert_array_equal)_check_n_features validate_datac@seZdZdddZdS) MyEstimatorrNcCs||_||_dSN)l1empty)selfr r!r#=E:\soccer\ml_env\Lib\site-packages\sklearn/tests/test_base.py__init__.szMyEstimator.__init__)rN__name__ __module__ __qualname__r%r#r#r#r$r-src@seZdZdddZdS)KNcCs||_||_dSr)cd)r"r+r,r#r#r$r%4sz K.__init__)NNr&r#r#r#r$r*3sr*c@seZdZdddZdS)TNcCs||_||_dSr)ab)r"r.r/r#r#r$r%:sz T.__init__)NNr&r#r#r#r$r-9sr-cseZdZfddZZS)NaNTagcst}d|j_|S)NTsuper__sklearn_tags__ input_tags allow_nanr"tags __class__r#r$r3@s zNaNTag.__sklearn_tags__r'r(r)r3 __classcell__r#r#r8r$r0?sr0cseZdZfddZZS)NoNaNTagcst}d|j_|SNFr1r6r8r#r$r3Gs zNoNaNTag.__sklearn_tags__r:r#r#r8r$r<Fsr<cseZdZfddZZS) OverrideTagcst}d|j_|Sr=r1r6r8r#r$r3Ns zOverrideTag.__sklearn_tags__r:r#r#r8r$r>Msr>c@s eZdZdS)DiamondOverwriteTagNr'r(r)r#r#r#r$r?Tsr?c@s eZdZdS)InheritDiamondOverwriteTagNr@r#r#r#r$rAXsrAc@s$eZdZdZedgfddZdS)ModifyInitParamsz_Deprecated behavior. Equal parameters but with a type cast. Doesn't fulfill a is a rcCs||_dSr)copyr.r"r.r#r#r$r%bszModifyInitParams.__init__N)r'r(r)__doc__nparrayr%r#r#r#r$rB\srBc@seZdZdZdddZdS)Buggyz9A buggy estimator that does not set its parameters right.NcCs d|_dS)Nr.rDr#r#r$r%iszBuggy.__init__)Nr'r(r)rEr%r#r#r#r$rHfsrHc@s(eZdZddZdddZd ddZdS) NoEstimatorcCsdSrr#r"r#r#r$r%nszNoEstimator.__init__NcCs|Srr#r"Xyr#r#r$fitqszNoEstimator.fitcCsdSrr#r"rOr#r#r$predicttszNoEstimator.predict)NN)N)r'r(r)r%rQrSr#r#r#r$rLms rLc@seZdZdZddZdS) VargEstimatorz-scikit-learn estimators shouldn't have vargs.cGsdSrr#)r"Zvargsr#r#r$r%{szVargEstimator.__init__NrKr#r#r#r$rTxsrTcCsnddlm}m}||dd}t|}||us0J||ksDJ||tdd}t|}||usjJdS)Nr SelectFpr f_classif皙?alpha) )sklearn.feature_selectionrVrWr get_paramsrFZzerosrVrWselectorZ new_selectorr#r#r$ test_clones  racCs<ddlm}m}||dd}d|_t|}t|dr8JdS)NrrUrXrYtest own_attribute)r]rVrWrcrhasattrr_r#r#r$ test_clone_2s  recCst}d|_ttt|Wdn1s40Yt}ttt|Wdn1sl0Yt}ttt|Wdn1s0Yt }ttt|Wdn1s0YdS)Nr\) rHr.pytestraises RuntimeErrorrrL TypeErrorrTrB)ZbuggyZ no_estimatorZvarg_estestr#r#r$test_clone_buggys & & & rkcCs^ttgd}t|}t|j|jtttdggd}t|}t|jj|jjdS)Nr!r) rrFrGrrr!spZ csr_matrixdataclfZclf2r#r#r$test_clone_empty_arrays rqcCs(ttjd}t|}|j|jus$JdSNrl)rrFnanrr!ror#r#r$test_clone_nans rtcCs*dti}t|}|d|dus&JdS)Nr.)rr)origZclonedr#r#r$test_clone_dicts rvcsnfddttD}|D]Ntd}t|d}t|}|jj|jjusRJt|j |j qdS)Ncs0g|](}|drttt|turqS)Z_matrix)endswithtypegetattrrm).0nameclsr#r$ s z.test_clone_sparse_matrices..rl) dirrmrFeyerrr!r9rZtoarray)Zsparse_matrix_classesZ sparse_matrixrpZ clf_clonedr#r|r$test_clone_sparse_matricess  rcCs&ttd}t|}|j|jus"JdSrr)rrr!ror#r#r$test_clone_estimator_typess rcCs>d}tjt|dttWdn1s00YdS)Nz8You should provide an instance of scikit-learn estimatormatch)rfrgrirr)msgr#r#r$%test_clone_class_rather_than_instancesrcCsTt}t|ttt}t|dks,Jtdgdd}tt|dksPJdS)NzT(a=K(), b=K())Z long_paramsirJi)rreprr-r*len) my_estimatorrbZsome_estr#r#r$ test_reprs rcCst}t|dSr)rstr)rr#r#r$test_strsrcCsttt}d|jddvs Jd|jddvs4J|jdd|jjdksPJtt|jddWdn1s|0YdS)Na__dT)deepFr\)r)Za__a) r-r*r^ set_paramsr.r,rfrg ValueError)rbr#r#r$test_get_paramss   rcCstjtddttsJWdn1s00YtjtddttsVJWdn1sj0YtjtddttsJWdn1s0Ytjtddt t sJWdn1s0YdS)Nz!passing a class to.*is deprecatedr) rfwarns FutureWarningr rr rr r r rr#r#r#r$test_is_estimator_type_classs***rzestimator, expected_resultTCrXrIsvcZsvc_cvFZsvrZsvr_cvcCst||ksJdSr)r  estimatorZexpected_resultr#r#r$test_is_classifiersrcCst||ksJdSr)r rr#r#r$test_is_regressor/srZ n_clusterskmZkm_cvcCst||ksJdSr)r rr#r#r$test_is_clusterer@srcCstdtfg}tt|jddWdn1s<0Ytt|jddWdn1sr0YdS)NrT)Zsvc__stupid_param)Zsvm__stupid_param)rrrfrgrr)rpr#r#r$test_set_paramsQs  * rcsRGfdddt}dddtd|fgt|ifD]}|jdddq:dS)NcseZdZfddZZS)z?test_set_params_passes_all_parameters..TestDecisionTreecs"tjfi||ksJ|Sr)r2r)r"kwargs)r9expected_kwargsr#r$rhs zJtest_set_params_passes_all_parameters..TestDecisionTree.set_params)r'r(r)rr;r#rr8r$TestDecisionTreegsrrr\) max_depthZmin_samples_leafr)Zestimator__max_depthZestimator__min_samples_leaf)rrrr)rrjr#rr$%test_set_params_passes_all_parameterscs   rcCs0tti}|jtdd|jjdks,JdS)NgE@)rZ estimator__C)rrrrrr)Zgscvr#r#r$$test_set_params_updates_valid_paramsvs rz tree,datasetr\)r random_state)rc Cshtjd}|\}}||||jddt|d}|||}|j|||d}d}||ksdJ|dS)NrrIr[)size) sample_weightz5Unweighted and weighted scores are unexpectedly equal)rFrandomZ RandomStaterQrandintrscore) treeZdatasetrngrOrPrZscore_unweightedZscore_weightedrr#r#r$test_score_sample_weight~s   rcCsbGdddtt}td}t|}||dd}t|}|j|jkjsNJ|j |j ks^JdS)Nc@s,eZdZdZd ddZd ddZdd ZdS) z3test_clone_pandas_dataframe..DummyEstimatora,This is a dummy class for generating numerical features This feature extractor extracts numerical features from pandas data frame. Parameters ---------- df: pandas data frame The pandas data frame parameter. Notes ----- NrIcSs||_||_dSr)df scalar_param)r"rrr#r#r$r%sz.DummyEstimator.__init__cSsdSrr#rNr#r#r$rQsz7test_clone_pandas_dataframe..DummyEstimator.fitcSsdSrr#rRr#r#r$ transformsz=test_clone_pandas_dataframe..DummyEstimator.transform)NrI)N)r'r(r)rEr%rQrr#r#r#r$DummyEstimators  rr[rI)r) rrrFZarangerrrvaluesallr)rr,reZcloned_er#r#r$test_clone_pandas_dataframes  rcCsGdddt}tddgddgddgg}t|}|j}||}t|j|t||t ddgddgd dgg}||t|j|| |t|j|t |}||usJt|j|d S) z:Checks that clone works with `__sklearn_clone__` protocol.c@s4eZdZddZddZddZddZd d Zd S) z,test_clone_protocol..FrozenEstimatorcSs ||_dSr)fitted_estimator)r"rr#r#r$r%sz5test_clone_protocol..FrozenEstimator.__init__cSs t|j|Sr)ryr)r"r{r#r#r$ __getattr__sz8test_clone_protocol..FrozenEstimator.__getattr__cSs|Srr#rMr#r#r$__sklearn_clone__sz>test_clone_protocol..FrozenEstimator.__sklearn_clone__c_s|Srr#r"argsrr#r#r$rQsz0test_clone_protocol..FrozenEstimator.fitc_s|jj|i|Sr)rrrr#r#r$ fit_transformsz:test_clone_protocol..FrozenEstimator.fit_transformN)r'r(r)r%rrrQrr#r#r#r$FrozenEstimators rr\rrIN) rrFrGrrQZ components_rrZget_feature_names_outZasarrayrr)rrOZpca componentsZ frozen_pcaZX_newZclone_frozen_pcar#r#r$test_clone_protocols       rcCst}t|j|j}t|}d|vs0Jt $t dt |}Wdn1sb0Y| |j|j}| |j|j}||ksJdS)N_sklearn_versionerror) r load_irisrrQrntargetpickledumpswarningscatch_warnings simplefilterloadsr)irisrZ tree_pickleZ tree_restoredZscore_of_originalZscore_of_restoredr#r#r$?test_pickle_version_warning_is_not_raised_with_matching_versions    (rc@seZdZddZdS)TreeBadVersioncCst|jddS)N something)_sklearn_version)dict__dict__itemsrMr#r#r$ __getstate__szTreeBadVersion.__getstate__Nr'r(r)rr#r#r#r$rsrzTrying to unpickle estimator {estimator} from version {old_version} when using version {current_version}. This might lead to breaking code or invalid results. Use at your own risk.cCst}t|j|j}t|}tj ddt j d}t j t|d}t|Wdn1sd0Y|jdj}t|tsJ|jdksJ|jdksJ|jt j ksJdS)NrrrZ old_versioncurrent_versionrr)rrrrQrnrrrpickle_error_messageformatsklearn __version__rfr UserWarningrlistmessage isinstancerZestimator_nameZoriginal_sklearn_versionZcurrent_sklearn_version)rrZtree_pickle_otherrZwarning_recordr#r#r$r?rA)Z nan_tag_estZno_nan_tag_estZredefine_tags_estZdiamond_tag_estZinherit_diamond_tag_estr#r#r$test_tag_inheritancesrcCsTGdddt}|}d}tjt|d|Wdn1sF0YdS)Nc@s eZdZdddZdddZdS) z.MyEstimatorrcSsdSrr#)r"paramr#r#r$r%szEtest_raises_on_get_params_non_attribute..MyEstimator.__init__NcSs|Srr#rNr#r#r$rQsz@test_raises_on_get_params_non_attribute..MyEstimator.fit)r)N)r'r(r)r%rQr#r#r#r$rs rz-'MyEstimator' object has no attribute 'param'r)rrfrgAttributeErrorr^)rrjrr#r#r$'test_raises_on_get_params_non_attributes rcCstt}|}d|vsJd|vs&Jtdd0|}d|vsFJd|vsRJWdn1sf0YdS)Nz text/plainz text/htmlrdisplay)r_repr_mimebundle_r)routputr#r#r$test_repr_mimebundle_s    rc Cst}|}d|vsJtddJd}tjt|d|}Wdn1sV0YWdn1st0YdS)Nz