Skip to content

Commit 7572ba1

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 0d800b1 commit 7572ba1

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
@@ -66,6 +66,15 @@ class NeighbourhoodCleaningRule(BaseCleaningSampler):
6666
:class:`sklearn.neighbors.base.KNeighborsMixin` that will be used to
6767
find the nearest-neighbors.
6868
69+
threshold_cleaning : float, optional (default=0.5)
70+
Threshold used to whether consider a class or not during the cleaning
71+
after applying ENN. A class will be considered during cleaning when:
72+
73+
Ci > C x T ,
74+
75+
where Ci and C is the number of samples in the class and the data set,
76+
respectively and theta is the threshold.
77+
6978
n_jobs : int, optional (default=1)
7079
The number of threads to open if possible.
7180

0 commit comments

Comments
 (0)