]jASSKJr SSKrSSKJr SSKJr SSK J r SSK J r Sr Sr\ "S S /S /S .S S 9S5rSrSrg))suppressN)sparse) is_scalar_nan)validate_params)_object_dtype_isnanc[[[5 SSKnXRLaUR U5sSSS5 $SSS5 [ U5(aURRS:Xa[R"U5nU$URRS;a%[R"UR[S9nU$[U5nU$X:HnU$!,(df  N=f)Nrf)iudtype)r ImportErrorAttributeErrorpandasNAisnarr kindnpisnanzerosshapeboolr)X value_to_maskrXts P/home/mande/repo/quber/.venv/lib/python3.13/site-packages/sklearn/utils/_mask.py_get_dense_maskrs +~ .  II %;;q> / . & /]## 77<<3 !B IWW\\Z '!''.B I %Q'B I  I) / .s #C  C.cn[R"U5(d [X5$[URU5nURS:Xa[R O[R nU"X RR5URR54UR[S9nU$)aCompute the boolean mask X == value_to_mask. Parameters ---------- X : {ndarray, sparse matrix} of shape (n_samples, n_features) Input data, where ``n_samples`` is the number of samples and ``n_features`` is the number of features. value_to_mask : {int, float} The value which is to be masked in X. Returns ------- X_mask : {ndarray, sparse matrix} of shape (n_samples, n_features) Missing mask. csr)rr ) spissparserdataformat csr_matrix csc_matrixindicescopyindptrrr)rrrsparse_constructor Xt_sparses r _get_maskr+&s" ;;q>>q00  /B*+((e*;" YY^^ qxx}}/qwwdI z array-likez sparse matrix)rmaskT)prefer_skip_nested_validationc[R"U5n[R"UR[R5(aU$[ US5(a'[R "URS5nX!nU$)aReturn a mask which is safe to use on X. Parameters ---------- X : {array-like, sparse matrix} Data on which to apply mask. mask : array-like Mask to be used on X. Returns ------- mask : ndarray Array that is safe to use on X. Examples -------- >>> from sklearn.utils import safe_mask >>> from scipy.sparse import csr_matrix >>> data = csr_matrix([[1], [2], [3], [4], [5]]) >>> condition = [False, True, True, False, True] >>> mask = safe_mask(data, condition) >>> data[mask].toarray() array([[2], [3], [5]]) toarrayr)rasarray issubdtyper signedintegerhasattraranger)rr-inds r safe_maskr7FsbF ::d D }}TZZ!1!122 q)ii 1 &y Kr,cxUS:waU[X5SS24$[R"SURS4S9$)a6Return a mask which is safer to use on X than safe_mask. This mask is safer than safe_mask since it returns an empty array, when a sparse matrix is sliced with a boolean mask with all False, instead of raising an unhelpful error in older versions of SciPy. See: https://github.com/scipy/scipy/issues/5361 Also note that we can avoid doing the dot product by checking if the len_mask is not zero in _huber_loss_and_gradient but this is not going to be the bottleneck, since the number of outliers and non_outliers are typically non-zero and it makes the code tougher to follow. Parameters ---------- X : {array-like, sparse matrix} Data on which to apply mask. mask : ndarray Mask to be used on X. len_mask : int The length of the mask. Returns ------- mask : ndarray Array that is safe to use on X. rN)r)r7rrr)rr-len_masks raxis0_safe_slicer;ss<@1}1#Q&'' 881aggaj/ **r,cU[R"U5::a [S5e[R"U[S9nSX 'U$)aConvert list of indices to boolean mask. Parameters ---------- indices : list-like List of integers treated as indices. mask_length : int Length of boolean mask to be generated. This parameter must be greater than max(indices). Returns ------- mask : 1d boolean nd-array Boolean array that is True where indices are present, else False. Examples -------- >>> from sklearn.utils._mask import indices_to_mask >>> indices = [1, 2 , 3, 4] >>> indices_to_mask(indices, 5) array([False, True, True, True, True]) z-mask_length must be greater than max(indices)r T)rmax ValueErrorrr)r& mask_lengthr-s rindices_to_maskr@s<.bffWo%HII 88Kt ,DDM Kr,) contextlibrnumpyrscipyrr sklearn.utils._missingrsklearn.utils._param_validationrsklearn.utils.fixesrrr+r7r;r@r,rrHs` 0;30@O ,#' ##L"+Jr,