Skip to content

Commit afe2132

Browse files
committed
DOC add the fitted attribute to SMOTENC
1 parent ec6fe7e commit afe2132

File tree

1 file changed

+30
-0
lines changed
  • imblearn/over_sampling/_smote

1 file changed

+30
-0
lines changed

imblearn/over_sampling/_smote/base.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,36 @@ class SMOTENC(SMOTE):
405405
algorithm. From now on, you can pass an estimator where `n_jobs` is
406406
already set instead.
407407
408+
Attributes
409+
----------
410+
sampling_strategy_ : dict
411+
Dictionary containing the information to sample the dataset. The keys
412+
corresponds to the class labels from which to sample and the values
413+
are the number of samples to sample.
414+
415+
nn_k_ : estimator object
416+
Validated k-nearest neighbours created from the `k_neighbors` parameter.
417+
418+
ohe_ : :class:`~sklearn.preprocessing.OneHotEncoder`
419+
The one-hot encoder used to encode the categorical features.
420+
421+
categorical_features_ : ndarray of shape (n_cat_features,), dtype=np.int64
422+
Indices of the categorical features.
423+
424+
continuous_features_ : ndarray of shape (n_cont_features,), dtype=np.int64
425+
Indices of the continuous features.
426+
427+
median_std_ : float
428+
Median of the standard deviation of the continuous features.
429+
430+
n_features_ : int
431+
Number of features observed at `fit`.
432+
433+
n_features_in_ : int
434+
Number of features in the input dataset.
435+
436+
.. versionadded:: 0.9
437+
408438
See Also
409439
--------
410440
SMOTE : Over-sample using SMOTE.

0 commit comments

Comments
 (0)