o &ѹg6@sddlZddlZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl m Z ddl m Z ddlmZmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZm Z m!Z!dd l"m#Z#ddl$m%Z%m&Z&e'#e(de)ej*+e j,gZ-e.ddede-dDZ/Wdn1swYgdZ0gdZ1ddZ2ddZ3ddZ4ddZ5e j67de j68ded d!Z9d"d#Z:e!d$d%Z;e!d&d'Z0s  rzsklearn.)prefixpath)z%sklearn.utils.deprecation.load_mlcompzsklearn.pipeline.make_pipelinezsklearn.pipeline.make_unionz%sklearn.utils.extmath.safe_sparse_dotzsklearn.utils._joblibZHalfBinomialLoss)fitZscoreZ fit_predictZ fit_transformZ partial_fitZpredictc sPtjdddddlm}g}tD]drqdkrqtjdd  t }Wdn1s5wYt |t j }d d |D}|D]\}}g}|t vsY|d rZqJt |r`qJtjdd  }||}Wdn1svwYt|rtd ||dft|jrqJ|t|j|7}|jD]0} t|| } t| rqd} | tvrt| } d| jvr| jdjdurdg} t| | d} || 7}q||7}qJt |t j}fdd |D}|D]/\}}|d rq|dkrdrqt|tfddt Dst|s|t|7}qqd |}t|dkr&t!d|dS)Nnumpydocz+numpydoc is required to test the docstrings)reasonr docscrapez .conftestzsklearn.utils.fixesT)recordcSs g|] }|djdr|qS)rsklearn) __module__ startswith)rclsrrrrds z-test_docstring_parameters.._z"Error for __init__ of %s in %s: %sy)rcsg|] }|djkr|qS)r)r&)rfnnamerrrs configurationsetupc3s|]}|vVqdS)Nr)rd)name_rr sz,test_docstring_parameters.. zDocstring Error: )"pytest importorskipr r#PUBLIC_MODULESendswithwarningscatch_warnings importlib import_moduleinspect getmembersisclass_DOCSTRING_IGNORESr' isabstractClassDoclen RuntimeErrorr__new__r__init__methodsgetattr_METHODS_IGNORE_NONE_Yr parametersdefault isfunctionranyjoinAssertionError)r#Z incorrectmoduleclassescnamer(Zthis_incorrectwZcdoc method_namemethodZ param_ignoresigresultZ functionsfnamefuncmsgr)r-r1rtest_docstring_parametersMsz                  rZcCs|tdddgiS)NCg?rr )ZSearchCVrrr_construct_searchcv_instancesr\cCs\|jdkr|ddddgfgdS|jdkr|dtfgd S|jd kr,|dtfgd SdS) NColumnTransformerZ transformer passthroughrr)Z transformersPipelineZclf)Zsteps FeatureUnion)Ztransformer_list)__name__r r ) Estimatorrrr$_construct_compose_pipeline_instances   rccCs8tjgdgdgdgdgdgtjd}||dS)N)rrr)rd)rrr)rrr)rrer)Zdtype) dictionary)nparrayZfloat64)rbrfrrr_construct_sparse_coders   riz-ignore::sklearn.exceptions.ConvergenceWarningzname, Estimatorc Cstdddlm}||}|d}|jdvrt|}n2|jdvr(t|}n(|jdkr2t|}n|jdkrJt d d dd \}}|t ||}nt t |}|jd kr\|jd dn5|jdkrh|jddn)|jdkss|jdrz|jddn|jdvr|jd dn |jdkr|jd dd|vr|jd d|jdkr|jddd|vr|jddi}|jdr|jdvrgd }n|jd!krdd d"d#dd$g}d}nt d d#dd d d%\}}t||}t||}|jjr| |n+|jjr | tj||fn|jjr| |tjd&f|n| |||D]:} | j|vr0q&d'| j } d(| vr?q&t!t"d)t#|| jsNJWdq&1sZwYq&t$|} d*d+|D} t%| &| } t%| &|} | rt'd,|jd-| dS).Nr rr"Z Attributes)ZHalvingRandomSearchCVZRandomizedSearchCVZHalvingGridSearchCVZ GridSearchCV)r]r_r`Z SparseCoderZFrozenEstimator) n_samples n_features random_stateZ SelectKBestre)kZDummyClassifierZ stratified)ZstrategyZCCAZPLSr)Z n_components)ZGaussianRandomProjectionZSparseRandomProjectionZTSNE)Z perplexitymax_iter)rprn)rnZ Vectorizer)ZCountVectorizerZHashingVectorizerZTfidfVectorizer)zThis is the first document.z%This document is the second document.zAnd this is the third one.zIs this the first document?ZDictVectorizer)foobar)rrZbaz)rlrmZ n_redundantZ n_classesrn. zonly categorycSsg|]}|jqSrr,)rattrrrrr%sz1test_fit_docstring_attributes..zUndocumented attributes for z: )(r4r5r r#rArar\rcrirr rnextrZ set_paramsr' get_paramsr7rrZ__sklearn_tags__Z target_tagsZ one_d_labelsZ two_d_labelsrgZc_Z input_tagsZ three_d_arrayZnewaxisr-rMdesclowerr FutureWarninghasattr_get_all_fitted_attributesset differencerN)r-rbr#doc attributesZestXr*Zskipped_attributesrxr{fit_attrZfit_attr_namesZundocumented_attrsrrrtest_fit_docstring_attributess                                  $rc Cst|j}t;tjdtdt|jD]&}t |j|}t |t s&qzt ||Wn t tfy8Yqw| |qWdn1sIwYdd|DS)zBGet all the fitted attributes of an estimator including propertieserrorrvNcSs$g|]}|dr|ds|qS)r))r7r')rrorrrrDs$z._get_all_fitted_attributes..)list__dict__keysr8r9filterwarningsr}dir __class__rG isinstancepropertyAttributeErrorappend)Z estimatorrr-objrrrr.s     rcCsLtjtjtjtjtjg}t|dddgdd}t|dgd|ddS) z>Check docstrings parameters of related metrics are consistent.TZaverageZ zero_division)include_paramsZexclude_paramsaThis parameter is required for multiclass/multilabel targets\. If ``None``, the metrics for each class are returned\. Otherwise, this determines the type of averaging performed on the data: ``'binary'``: Only report results for the class specified by ``pos_label``\. This is applicable only if targets \(``y_\{true,pred\}``\) are binary\. ``'micro'``: Calculate metrics globally by counting the total true positives, false negatives and false positives\. ``'macro'``: Calculate metrics for each label, and find their unweighted mean\. This does not take label imbalance into account\. ``'weighted'``: Calculate metrics for each label, and find their average weighted by support \(the number of true instances for each label\)\. This alters 'macro' to account for label imbalance; it can result in an F-score that is not between precision and recall\.[\s\w]*\.* ``'samples'``: Calculate metrics for each instance, and find their average \(only meaningful for multilabel classification where this differs from :func:`accuracy_score`\)\.ru)rZdescr_regex_patternN) rZprecision_recall_fscore_supportZf1_scoreZ fbeta_scoreZprecision_scoreZ recall_scorerrMsplit)Zmetrics_to_checkZdescription_regexrrr3test_precision_recall_f_score_docstring_consistencyGs$  rcCstttggdddgddS)z?Check docstrings parameters stacking estimators are consistent.)cvZn_jobsr^verboseTZfinal_estimator_)rZ include_attrs exclude_attrsN)rrrrrrr8test_stacking_classifier_regressor_docstring_consistencyxs  r)=r:r<osr8rpkgutilrZnumpyrgr4r%rZsklearn.datasetsrZsklearn.ensemblerrZsklearn.experimentalrr Zsklearn.linear_modelr Zsklearn.preprocessingr Z sklearn.utilsr Z-sklearn.utils._test_common.instance_generatorrZsklearn.utils._testingrrrrrZsklearn.utils.deprecationrZsklearn.utils.estimator_checksrrr9 simplefilterr}rdirname__file__Z sklearn_pathrr6r?rHrZr\rcrimarkrZ parametrizerrrrrrrrsV              O  y 0