Skip to content

Commit 0f24ab4

Browse files
committed
DOC add threshold_cleaning in NCR (#335)
* DOC add threshold_cleaning parameter in the docstring * DOC remove useless line * DOC plain text instead of latex * DOC plain text instead of latex
1 parent 2b0c106 commit 0f24ab4

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

imblearn/under_sampling/prototype_selection/neighbourhood_cleaning_rule.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,15 @@ class NeighbourhoodCleaningRule(BaseCleaningSampler):
7373
:class:`sklearn.neighbors.base.KNeighborsMixin` that will be used to
7474
find the nearest-neighbors.
7575
76+
threshold_cleaning : float, optional (default=0.5)
77+
Threshold used to whether consider a class or not during the cleaning
78+
after applying ENN. A class will be considered during cleaning when:
79+
80+
Ci > C x T ,
81+
82+
where Ci and C is the number of samples in the class and the data set,
83+
respectively and theta is the threshold.
84+
7685
n_jobs : int, optional (default=1)
7786
The number of threads to open if possible.
7887

0 commit comments

Comments
 (0)