o ?Hh6@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._joblibHalfBinomialLoss)fitscore fit_predict fit_transform partial_fitpredictc 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* incorrectmoduleclassescnamer/this_incorrectwcdoc method_namemethod param_ignoresigresult functionsfnamefuncmsgr)r4r8rtest_docstring_parametersMsz                  rfcCs|tdddgiS)NCg?rr )SearchCVrrr_construct_searchcv_instancesricCs\|jdkr|ddddgfgdS|jdkr|dtfgd S|jd kr,|dtfgd SdS) NColumnTransformer transformer passthroughrr) transformersPipelineclf)steps FeatureUnion)transformer_list)__name__r r ) Estimatorrrr$_construct_compose_pipeline_instances   rucCs8tjgdgdgdgdgdgtjd}||dS)N)rrr)rv)rrr)rrr)rrwr)dtype) dictionary)nparrayfloat64)rtryrrr_construct_sparse_coders   r}z-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]9} | j|vr0q&d'| j } d(| vr?q&t!t"d)t#|| jsNJWdn 1sYwYq&t$|} d*d+|D} t%| &| } t%| &|} | rt'd,|jd-| dS).Nr'rr) Attributes)HalvingRandomSearchCVRandomizedSearchCVHalvingGridSearchCV GridSearchCV)rjrnrq SparseCoderFrozenEstimator) n_samples n_features random_state SelectKBestrw)kDummyClassifier stratified)strategyCCAPLSr) n_components)GaussianRandomProjectionSparseRandomProjectionTSNE) perplexitymax_iter)rr)r Vectorizer)CountVectorizerHashingVectorizerTfidfVectorizer)zThis is the first document.z%This document is the second document.zAnd this is the third one.zIs this the first document?DictVectorizer)foobar)rbaz)rr n_redundant n_classesr. zonly categorycSsg|]}|jqSrr3)rattrrrrr%sz1test_fit_docstring_attributes..zUndocumented attributes for z: )(r;r<r'r*rHrsrirur}rr r!nextr set_paramsr. get_paramsr>rr__sklearn_tags__ target_tags one_d_labels two_d_labelsrzc_ input_tags three_d_arraynewaxisr4rTdesclowerr FutureWarninghasattr_get_all_fitted_attributesset differencerU)r4rtr*doc attributesestXr1skipped_attributesrrfit_attrfit_attr_namesundocumented_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 propertieserrorrNcSs$g|]}|dr|ds|qS)r0)r>r.)rrrrrrDs$z._get_all_fitted_attributes..)list__dict__keysr?r@filterwarningsrdir __class__rN isinstancepropertyAttributeErrorappend) estimatorrr4objrrrr.s     rcCsLtjtjtjtjtjg}t|dddgdd}t|dgd|ddS) z>Check docstrings parameters of related metrics are consistent.Taverage zero_division)include_paramsexclude_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`\)\.r)rdescr_regex_patternN) rprecision_recall_fscore_supportf1_score fbeta_scoreprecision_score recall_scorerrTsplit)metrics_to_checkdescription_regexrrr3test_precision_recall_f_score_docstring_consistencyGs$  rcCstttggdddgddS)z?Check docstrings parameters stacking estimators are consistent.)cvn_jobsrlverboseTfinal_estimator_)r include_attrs exclude_attrsN)rrrrrrr8test_stacking_classifier_regressor_docstring_consistencyxs  r)=rArCosr?rpkgutilrnumpyrzr;r,rsklearn.datasetsrsklearn.ensemblerrsklearn.experimentalrr sklearn.linear_modelr sklearn.preprocessingr sklearn.utilsr -sklearn.utils._test_common.instance_generatorrsklearn.utils._testingrrrrrsklearn.utils.deprecationrsklearn.utils.estimator_checksrrr@ simplefilterrrdirname__file__ sklearn_pathrr=rFrOrfrirur}markr parametrizerrrrrrrrsV              O  y 0