Skip to content

Commit 0238083

Browse files
committed
DOC document kind_sel in NCR
1 parent f7e477d commit 0238083

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

imblearn/under_sampling/_prototype_selection/_neighbourhood_cleaning_rule.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ class NeighbourhoodCleaningRule(BaseCleaningSampler):
3838
:class:`sklearn.neighbors.base.KNeighborsMixin` that will be used to
3939
find the nearest-neighbors.
4040
41+
kind_sel : str, optional (default='all')
42+
Strategy to use in order to exclude samples in the ENN sampling.
43+
44+
- If ``'all'``, all neighbours will have to agree with the samples of
45+
interest to not be excluded.
46+
- If ``'mode'``, the majority vote of the neighbours will be used in
47+
order to exclude a sample.
48+
4149
threshold_cleaning : float, optional (default=0.5)
4250
Threshold used to whether consider a class or not during the cleaning
4351
after applying ENN. A class will be considered during cleaning when:

0 commit comments

Comments
 (0)