-iQSSKrSSKJr SSKrSSKJrJrJr SSK J r SSK J r J r Jr SSKJrJr SSKJrJrJrJrJr S S KJr "S S \\5rg) N)Integral) BaseEstimatorTransformerMixin _fit_context)resample)IntervalOptions StrOptions)_averaged_weighted_percentile_weighted_percentile)_check_feature_names_in_check_sample_weight check_arraycheck_is_fitted validate_data) OneHotEncoderc (\rSrSr%Sr\"\SSSS9S/\"1Sk5/\"1S k5/\"1S k5/\"\ \ R\ R15S/\"\S SSS9S/S /S .r \\S'SSSSSSSS.Sjjr\"SS9SSj5rSrSrSrSSjrSrg) KBinsDiscretizera6 Bin continuous data into intervals. Read more in the :ref:`User Guide `. .. versionadded:: 0.20 Parameters ---------- n_bins : int or array-like of shape (n_features,), default=5 The number of bins to produce. Raises ValueError if ``n_bins < 2``. encode : {'onehot', 'onehot-dense', 'ordinal'}, default='onehot' Method used to encode the transformed result. - 'onehot': Encode the transformed result with one-hot encoding and return a sparse matrix. Ignored features are always stacked to the right. - 'onehot-dense': Encode the transformed result with one-hot encoding and return a dense array. Ignored features are always stacked to the right. - 'ordinal': Return the bin identifier encoded as an integer value. strategy : {'uniform', 'quantile', 'kmeans'}, default='quantile' Strategy used to define the widths of the bins. - 'uniform': All bins in each feature have identical widths. - 'quantile': All bins in each feature have the same number of points. - 'kmeans': Values in each bin have the same nearest center of a 1D k-means cluster. For an example of the different strategies see: :ref:`sphx_glr_auto_examples_preprocessing_plot_discretization_strategies.py`. quantile_method : {"inverted_cdf", "averaged_inverted_cdf", "closest_observation", "interpolated_inverted_cdf", "hazen", "weibull", "linear", "median_unbiased", "normal_unbiased"}, default="linear" Method to pass on to np.percentile calculation when using strategy="quantile". Only `averaged_inverted_cdf` and `inverted_cdf` support the use of `sample_weight != None` when subsampling is not active. .. versionadded:: 1.7 dtype : {np.float32, np.float64}, default=None The desired data-type for the output. If None, output dtype is consistent with input dtype. Only np.float32 and np.float64 are supported. .. versionadded:: 0.24 subsample : int or None, default=200_000 Maximum number of samples, used to fit the model, for computational efficiency. `subsample=None` means that all the training samples are used when computing the quantiles that determine the binning thresholds. Since quantile computation relies on sorting each column of `X` and that sorting has an `n log(n)` time complexity, it is recommended to use subsampling on datasets with a very large number of samples. .. versionchanged:: 1.3 The default value of `subsample` changed from `None` to `200_000` when `strategy="quantile"`. .. versionchanged:: 1.5 The default value of `subsample` changed from `None` to `200_000` when `strategy="uniform"` or `strategy="kmeans"`. random_state : int, RandomState instance or None, default=None Determines random number generation for subsampling. Pass an int for reproducible results across multiple function calls. See the `subsample` parameter for more details. See :term:`Glossary `. .. versionadded:: 1.1 Attributes ---------- bin_edges_ : ndarray of ndarray of shape (n_features,) The edges of each bin. Contain arrays of varying shapes ``(n_bins_, )`` Ignored features will have empty arrays. n_bins_ : ndarray of shape (n_features,), dtype=np.int64 Number of bins per feature. Bins whose width are too small (i.e., <= 1e-8) are removed with a warning. n_features_in_ : int Number of features seen during :term:`fit`. .. versionadded:: 0.24 feature_names_in_ : ndarray of shape (`n_features_in_`,) Names of features seen during :term:`fit`. Defined only when `X` has feature names that are all strings. .. versionadded:: 1.0 See Also -------- Binarizer : Class used to bin values as ``0`` or ``1`` based on a parameter ``threshold``. Notes ----- For a visualization of discretization on different datasets refer to :ref:`sphx_glr_auto_examples_preprocessing_plot_discretization_classification.py`. On the effect of discretization on linear models see: :ref:`sphx_glr_auto_examples_preprocessing_plot_discretization.py`. In bin edges for feature ``i``, the first and last values are used only for ``inverse_transform``. During transform, bin edges are extended to:: np.concatenate([-np.inf, bin_edges_[i][1:-1], np.inf]) You can combine ``KBinsDiscretizer`` with :class:`~sklearn.compose.ColumnTransformer` if you only want to preprocess part of the features. ``KBinsDiscretizer`` might produce constant features (e.g., when ``encode = 'onehot'`` and certain bins do not contain any data). These features can be removed with feature selection algorithms (e.g., :class:`~sklearn.feature_selection.VarianceThreshold`). Examples -------- >>> from sklearn.preprocessing import KBinsDiscretizer >>> X = [[-2, 1, -4, -1], ... [-1, 2, -3, -0.5], ... [ 0, 3, -2, 0.5], ... [ 1, 4, -1, 2]] >>> est = KBinsDiscretizer( ... n_bins=3, encode='ordinal', strategy='uniform' ... ) >>> est.fit(X) KBinsDiscretizer(...) >>> Xt = est.transform(X) >>> Xt # doctest: +SKIP array([[ 0., 0., 0., 0.], [ 1., 1., 1., 0.], [ 2., 2., 2., 1.], [ 2., 2., 2., 2.]]) Sometimes it may be useful to convert the data back into the original feature space. The ``inverse_transform`` function converts the binned data into the original feature space. Each value will be equal to the mean of the two bin edges. >>> est.bin_edges_[0] array([-2., -1., 0., 1.]) >>> est.inverse_transform(Xt) array([[-1.5, 1.5, -3.5, -0.5], [-0.5, 2.5, -2.5, -0.5], [ 0.5, 3.5, -1.5, 0.5], [ 0.5, 3.5, -1.5, 1.5]]) rNleft)closedz array-like>onehotordinal onehot-dense>kmeansuniformquantile> warnhazenlinearweibull inverted_cdfmedian_unbiasednormal_unbiasedclosest_observationaveraged_inverted_cdfinterpolated_inverted_cdfr random_staten_binsencodestrategyquantile_methoddtype subsampler*_parameter_constraintsrrr i@ )r-r.r/r0r1r*cXXlX lX0lX@lXPlX`lXplgNr+)selfr,r-r.r/r0r1r*s X/var/www/html/venv/lib/python3.13/site-packages/sklearn/preprocessing/_discretization.py__init__KBinsDiscretizer.__init__s)   . "(T)prefer_skip_nested_validationc  [XSS9nUR[R[R4;a URnO URnUR upVUb[ X1URS9nURb2XPR:a#[USURURUS9nSnUR SnURU5n[R"U[S9nURn URS:Xa#U S:Xa[R "S ["5 S n URS:XaU S ;aUb[%S U S 35eURS:wa UbUS:gn O ['S5n [)U5GHn USS2U 4n XR+5n XR-5nX:XaV[R "SU -5 SX{'[R."[R0*[R0/5X'MURS:Xa [R2"XX{S-5X'GOzURS:Xa[R2"SSX{S-5n0nU S :waUcU US'Uc;[R4"[R6"X40UD6[RS9X'O[8[:S .U n[R4"UVs/sH nU"XUS9PM sn[RS9X'OURS:XaSSKJn [R2"XX{S-5nUSSUSS-SS2S4S-nU"X{USS9nURAU SS2S4US9RBSS2S4nURE5 USSUSS-S-X'[RFXU U4X'URS;dGMI[RH"X[R0S9S:nXUX'[KX5S- X{:wdGM[R "SU -5 [KX5S- X{'GM Xl&Xpl'SURP;a[SURNVs/sHn[RT"U5PM snURPS:HUS 9Ul+URVRA[R"S[KURN5455 U$s snfs snf)!a Fit the estimator. Parameters ---------- X : array-like of shape (n_samples, n_features) Data to be discretized. y : None Ignored. This parameter exists only for compatibility with :class:`~sklearn.pipeline.Pipeline`. sample_weight : ndarray of shape (n_samples,) Contains weight values to be associated with each sample. .. versionadded:: 1.3 .. versionchanged:: 1.7 Added support for strategy="uniform". Returns ------- self : object Returns the instance itself. numericr0NT)replace n_samplesr* sample_weightrrr a%The current default behavior, quantile_method='linear', will be changed to quantile_method='averaged_inverted_cdf' in scikit-learn version 1.9 to naturally support sample weight equivalence properties by default. Pass quantile_method='averaged_inverted_cdf' explicitly to silence this warning.r")r$r(zWhen fitting with strategy='quantile' and sample weights, quantile_method should either be set to 'averaged_inverted_cdf' or 'inverted_cdf', got quantile_method='z ' instead.rz3Feature %d is constant and will be replaced with 0.rdmethod)percentile_rankrr)KMeans?) n_clustersinitn_init)r@)rr)to_beging:0yE>zqBins whose width are too small (i.e., <= 1e-8) in feature %d are removed. Consider decreasing the number of bins.r) categories sparse_outputr0),rr0npfloat64float32shaperr1rr*_validate_n_binszerosobjectr/r.warningsr FutureWarning ValueErrorslicerangeminmaxarrayinflinspaceasarray percentiler r clusterrDfitcluster_centers_sortr_ediff1dlen bin_edges_n_bins_r-rarange_encoder)r5Xyr@ output_dtyper? n_featuresr, bin_edgesr/nnz_weight_maskjjcolumncol_mincol_maxpercentile_levelspercentile_kwargspercentile_funcprD uniform_edgesrHkmcentersmaskis r6raKBinsDiscretizer.fits6 $ 3 ::"**bjj1 1::L77L !  $0QM >> %)nn*D..!..+ A!MWWQZ &&z2HHZv6 .. ==J &?f+D MM  'O MMZ ''PP)88G7H T  ==J &=+D,q0O$DkO #Bq"uXF-113G-113G! IBN "266'266): ; }} ) " Gfj1n M *,$&KK3 Q$G! %'!"h.=3H2A%h/ ($&JJ fUCTU jj%IM)=1N'&''O%'JJ&7%6,FSTU%6!jj %IM(*,!# Gfj1n M %ab)M#2,>>4H3NvzQG&&1d7O=!""1a4) !(ws|!;s B "g}g&E F }} 66zz)-"&&ADH ) d 3 y}%)VZ7MM9;=> "%Y]!3a!7FJU$X$ t{{ ")26,,?,QBIIaL,?"kkX5"DM MM  bhh3t||+<'=> ? aP@s S6 7 S;cURn[U[5(a[R"X[ S9$[ U[ SSS9nURS:dURSU:wa [S5eUS:X2:g-n[R"U5SnURSS:aAS RS U55n[S R[RU55eU$) z0Returns n_bins_, the number of bins per feature.r=TF)r0copy ensure_2drrz8n_bins must be a scalar or array of shape (n_features,).rz, c38# UHn[U5v M g7fr4)str).0r}s r6 4KBinsDiscretizer._validate_n_bins..sB0A1A0Aszk{} received an invalid number of bins at indices {}. Number of bins must be at least 2, and must be an int.)r, isinstancerrMfullintrndimrPrVwherejoinformatr__name__)r5rn orig_binsr,bad_nbins_valueviolating_indicesindicess r6rQ!KBinsDiscretizer._validate_n_binssKK i * *77:< <YcN ;;?fll1o;WX X!A:&*=>HH_5a8  " "1 % )iiB0ABBG::@&$--w;  r9cz[U5 URc [R[R4O URn[ XSUSS9nUR n[URS5H,n[R"XESSUSS2U4SS9USS2U4'M. URS :XaU$SnS UR;a1URRnURURlURRU5nX`RlU$!X`Rlf=f) a3 Discretize the data. Parameters ---------- X : array-like of shape (n_samples, n_features) Data to be discretized. Returns ------- Xt : {ndarray, sparse matrix}, dtype={np.float32, np.float64} Data in the binned space. Will be a sparse matrix if `self.encode='onehot'` and ndarray otherwise. NTF)rr0resetrrEright)siderr) rr0rMrNrOrrgrXrP searchsortedr-rj transform)r5rkr0Xtrorq dtype_initXt_encs r6rKBinsDiscretizer.transforms -1JJ,>RZZ(DJJ 4U% HOO  $B a(;R2YWUBq"uI% ;;) #I t{{ ",,J"$((DMM  -]],,R0F#-MM  #-MM s ;D((D:c&[U5 SUR;aURRU5n[ US[ R [ R4S9nURRSnURSU:wa'[SRX2RS55e[U5HOnURUnUSSUSS-S -nXbSS2U4R[ R5USS2U4'MQ U$) az Transform discretized data back to original feature space. Note that this function does not regenerate the original data due to discretization rounding. Parameters ---------- X : array-like of shape (n_samples, n_features) Transformed data in the binned space. Returns ------- X_original : ndarray, dtype={np.float32, np.float64} Data in the original feature space. rT)rr0rrz8Incorrect number of features. Expecting {}, received {}.NrErF)rr-rjinverse_transformrrMrNrOrhrPrVrrXrgastypeint64)r5rkXinvrnrqro bin_centerss r6r"KBinsDiscretizer.inverse_transforms$  t{{ " //2A14 BJJ/GH\\''* ::a=J &JQQ 1    #B+I$QR=9Sb>9S@K%ArE{&:&:288&DEDBK$  r9c[US5 [X5n[US5(aURR U5$U$)a\Get output feature names. Parameters ---------- input_features : array-like of str or None, default=None Input features. - If `input_features` is `None`, then `feature_names_in_` is used as feature names in. If `feature_names_in_` is not defined, then the following input feature names are generated: `["x0", "x1", ..., "x(n_features_in_ - 1)"]`. - If `input_features` is an array-like, then `input_features` must match `feature_names_in_` if `feature_names_in_` is defined. Returns ------- feature_names_out : ndarray of str objects Transformed feature names. n_features_in_rj)rrhasattrrjget_feature_names_out)r5input_featuress r6r&KBinsDiscretizer.get_feature_names_out sC( ./0F 4 $ $==66~F Fr9) rjrgr0r-r,rhr/r*r.r1))NNr4)r __module__ __qualname____firstlineno____doc__r rr r typerMrNrOr2dict__annotations__r7rrarQrrr__static_attributes__r9r6rrs]@Haf=|LCDE ABC    $RZZ 894@xD@$G'(-$D6))&5|6||2%N%Nr9r)rTnumbersrnumpyrMbaserrrutilsrutils._param_validationr r r utils.statsr r utils.validationrrrrr _encodersrrrr9r6rsB @@CCM%L'Lr9