a j68@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'Fe(de)ej*+e j,gZ-e.ddede-dDZ/Wdn1sF0YgdZ0gdZ1ddZ2ddZ3ddZ4ddZ5e j67de j68ded d!Z9d"d#Z:e!d$d%Z;e!d&d'Z0srzsklearn.)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 sntjdddddlm}g}tD] dr4q"dkr>q"tjdd t }Wdn1sj0Yt |t j }d d |D}|D]\}}g}|t vs|d rqt |rqtjdd }||}Wdn1s0Yt|rtd ||dft|jr&q|t|j|7}|jD]l} t|| } t| rXq._z"Error for __init__ of %s in %s: %sy)rcsg|]}|djkr|qS)r)r&)rfnnamerrrr) configurationsetupc3s|]}|vVqdS)Nr)rd)name_rr r)z,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.r2rtest_docstring_parametersMsp   ( (            r[cCs|tdddgiS)NCg?rr )ZSearchCVrrr_construct_searchcv_instancesr]cCs\|jdkr |ddddgfgdS|jdkr<|dtfgd S|jd krX|dtfgd SdS) NColumnTransformerZ transformer passthroughrr)Z transformersPipelineZclf)Zsteps FeatureUnion)Ztransformer_list)__name__r r ) Estimatorrrr$_construct_compose_pipeline_instances    rdcCs8tjgdgdgdgdgdgtjd}||dS)N)rrr)re)rrr)rrr)rrfr)Zdtype) dictionary)nparrayZfloat64)rcrgrrr_construct_sparse_coders  rjz-ignore::sklearn.exceptions.ConvergenceWarningzname, Estimatorc Cs$tdddlm}||}|d}|jdvr.zUndocumented attributes for z: )(r5r6r r#rBrbr]rdrjrr rnextrZ set_paramsr' get_paramsr8rrZ__sklearn_tags__Z target_tagsZ one_d_labelsZ two_d_labelsrhZc_Z input_tagsZ three_d_arrayZnewaxisr.rNdesclowerr FutureWarninghasattr_get_all_fitted_attributesset differencerO)r.rcr#doc attributesZestXr+Zskipped_attributesrzr}fit_attrZfit_attr_namesZundocumented_attrsrrrtest_fit_docstring_attributess                               6rc Cst|j}tztjdtdt|jD]P}t |j|}t |t sLq0zt ||Wnt tfytYq0Yn0| |q0Wdn1s0Ydd|DS)zBGet all the fitted attributes of an estimator including propertieserrorrxNcSs$g|]}|dr|ds|qS)r*)r8r')rrqrrrrDr)z._get_all_fitted_attributes..)list__dict__keysr9r:filterwarningsrdir __class__rH 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`\)\.rw)rZdescr_regex_patternN) rZprecision_recall_fscore_supportZf1_scoreZ fbeta_scoreZprecision_scoreZ recall_scorerrNsplit)Zmetrics_to_checkZdescription_regexrrr3test_precision_recall_f_score_docstring_consistencyGs$ rcCstttggdddgddS)z?Check docstrings parameters stacking estimators are consistent.)ZcvZn_jobsr_verboseTZfinal_estimator_)rZ include_attrsZ exclude_attrsN)rrrrrrr8test_stacking_classifier_regressor_docstring_consistencyxs r)=r;r=osr9rpkgutilrnumpyrhr5r%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_checksrrr: simplefilterrrdirname__file__Z sklearn_pathrr7r@rIr[r]rdrjmarkrZ parametrizerrrrrrrrsR            $   O   y 0