o i@s(ddlmZmZdddZdddZdS) )_find_matching_floating_dtypeget_namespace_and_device2Ncst|\}}t|d}j|||d}j|||d}|j}|dkr&|S|jdkr1|d}|j|jkrM|jd|jdkrM||jddfj}j|dd}j ||dd} |jd} ||dj | |df}   | rj  ||dd} d| | <j | jdd|d d dk} | | d| <fd d tjdD}|jdd}|d|}j |jd|d}|||f}|||f}|dkr|dS|S) a3Compute the weighted percentile with method 'inverted_cdf'. When the percentile lies between two data points of `array`, the function returns the lower value. If `array` is a 2D array, the `values` are selected along axis 0. `NaN` values are ignored by setting their weights to 0. If `array` is 2D, this is done in a column-isolated manner: a `NaN` in the second column, does not impact the percentile computed for the first column even if `sample_weight` is 1D. .. versionchanged:: 0.24 Accepts 2D `array`. .. versionchanged:: 1.7 Supports handling of `NaN` values. Parameters ---------- array : 1D or 2D array Values to take the weighted percentile of. sample_weight: 1D or 2D array Weights for each value in `array`. Must be same shape as `array` or of shape `(array.shape[0],)`. percentile_rank: int or float, default=50 The probability level of the percentile to compute, in percent. Must be between 0 and 100. xp : array_namespace, default=None The standard-compatible namespace for `array`. Default: infer. Returns ------- percentile : scalar or 0D array if `array` 1D (or 0D), array if `array` 2D Weighted percentile at the requested probability level. xp)Zdtypedevice)r )Zaxis)r .)rd).r cs$g|]}|df|qS).)Z searchsorted).0Z feature_idxZadjusted_percentile_rankZ weight_cdfrb/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/sklearn/utils/stats.py as z(_weighted_percentile..)rrZasarrayndimZreshapeshapeZtileTZargsortZtake_along_axisZarangeanyisnanZcumulative_sum nextafterstackrangeZclip)array sample_weightpercentile_rankr_rZfloating_dtypeZn_dimZ sorted_idxZsorted_weightsZ n_featuresZlargest_value_per_columnZsorted_nan_maskmaskZpercentile_indicesZmax_idxZ col_indicesZpercentile_in_sortedresultrr r_weighted_percentile sJ'          rcCs*t||||dt| |d||ddS)Nrr r)r)rrrrrrr_averaged_weighted_percentilevs r )rN)Zutils._array_apirrrr rrrrs l