o tBh @s<ddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZee e e e eeee d ZeeeddeedZgdZddgZgdZejdZejdddZejdddZ ddZ!ej"#dej"$dd d!eDd"d#Z%ej"$dd$d!eDd%d&Z&ej"#dej"$d'ed(d)Z'ej"#dej"$d'eeed*d+Z(ej"#dej"$d'eeed,d-Z)ej"$d.e*d/d0Z+ej"$d1e,efie-d2d3Z.dS)4)partial)chainN)adjusted_mutual_info_score)adjusted_rand_score) rand_score)completeness_score)fowlkes_mallows_score)homogeneity_score)mutual_info_score)normalized_mutual_info_score)v_measure_score)silhouette_score)calinski_harabasz_score)davies_bouldin_score)assert_allclose) rrrrr r r r r manhattan)metric)r silhouette_manhattanrr)rrr r rr rr r)rrr rr rrr sizecCstttttks JdSN)sortedSYMMETRIC_METRICSNON_SYMMETRIC_METRICSSUPERVISED_METRICSrr/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/metrics/cluster/tests/test_common.py"test_symmetric_non_symmetric_union]s rzignore::FutureWarningzmetric_name, y1, y2cCg|]}|ttfqSry1y2.0namerrr fr'cCs*t|}|||t|||ksJdSrrpytestapprox metric_namer"r#rrrr test_symmetryds"r.cCr rr!r$rrrr'nr(cCs*t|}|||t|||ksJdSrr)r,rrrtest_non_symmetryms"r/r-cCsgd}gd}t|}|gdgddksJ|gdgddks&J|gdgddks3J|gdgddks@J|||tdksLJgd }gd }t||||||g}|d kriJdS) N)rrrr0r0)rrrr0r0)rrrr0g)rrr0r0r1)rrr0r0r0)rr0r0r0r0g?)rrrrrr)rr0r1rr)rr*r+nparrayany)r- upper_bound_1 upper_bound_2r lower_bound_1 lower_bound_2scorerrrtest_normalized_outputvsr<cCstgd}tgd}|tvr=t|}|||}t||d||t||d|d|t|||d|dSt|}tjjddd}|||}t|||d|dS)N)rrrr0r0rr0)r0rr0rr0r0rr0 )r=r)r4r5rrUNSUPERVISED_METRICSrandomrandint)r-y_labely_predrscore_1Xrrrtest_permute_labelss  rFc Csgd}gd}dd}|tvr:t|}|||}||}||}t||D]\\}} \} } |||| ks7Jq&dSt|}tjjddd} || |}||| t|ksWJ||}|D] \}} ||| |ksjJq]dS)N)rrrrr0r0r0r0)rr0r1rr2r3r>css~t|}|dfV|dfVdd|DdfVtjdd|DtddfV|d d fV|d d fVdS) Nz array of intsz list of intscSg|]}t|dqSz-astrr%xrrrr'zDtest_format_invariance..generate_formats..z list of strscSrHrIrJrLrrrr'rN)dtypez array of strsr0zincluding negative intszstrictly positive ints)r4r5tolistobject)yrrrgenerate_formatss  z0test_format_invariance..generate_formatsr=)r=r)rzipr?r4r@rAastypefloat) r-y_truerCrSrrD y_true_gen y_pred_gen y_true_fmtfmt_name y_pred_fmt_rErrrtest_format_invariances&    r_rcCs dD] \}}||g|gqdS)N))rr)rr0)r0r)r0r0r)rijrrrtest_single_samples rbzmetric_name, metric_funccCs|tvr"ddgtjtjgfddgtjtjgfddgtjtjgfg}ntjjddd}|tjtjgf|tjtjgf|tjtjgfg}tjtdd|D]}||qKWddS1s]wYdS)Nrr0r=)r1r=rzcontains (NaN|infinity))match) rr4infnanr@rAr*raises ValueError)r- metric_funcinvalidsrEargsrrrtest_inf_nan_inputs. "rk)/ functoolsr itertoolsrr*numpyr4sklearn.metrics.clusterrrrrrr r r r r rrsklearn.utils._testingrrr?rrNORMALIZED_METRICSr@ RandomStaterngrAr"r#rmarkfilterwarnings parametrizer.r/r<rFr_valuesrbdictitemsrkrrrrs~                           !