Skip to content

Commit 6b5c9d9

Browse files
chkoarglemaitre
authored andcommitted
Remove UnbalancedDataset references (#127)
1 parent 2b8643f commit 6b5c9d9

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

CONTRIBUTING.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ GitHub:
1717

1818
2. Clone this copy to your local disk:
1919

20-
$ git clone [email protected]:YourLogin/UnbalancedDataset.git
20+
$ git clone [email protected]:YourLogin/imbalanced-learn.git
2121
$ cd UnbalancedDataset
2222

2323
3. Create a branch to hold your changes:

imblearn/under_sampling/nearmiss.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -212,9 +212,9 @@ def _sample(self, X, y):
212212

213213
# Assign the parameter of the element of this class
214214
# Check that the version asked is implemented
215-
if not (self.version == 1 or self.version == 2 or self.version == 3):
216-
raise ValueError('UnbalancedData.NearMiss: there is only 3 '
217-
'versions available with parameter version=1/2/3')
215+
if self.version not in [1,2,3]:
216+
raise ValueError("Parameter 'version' must be 1, 2 or 3, "
217+
"got {0}".format(self.version))
218218

219219
# Start with the minority class
220220
X_min = X[y == self.min_c_]

imblearn/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
__version__ = '0.2.0.dev0'
2525

2626
_IMBALANCED_DATASET_INSTALL_MSG = 'See %s for installation information.' % (
27-
'glemaitre.github.io/UnbalancedDataset/install.html')
27+
'http://contrib.scikit-learn.org/imbalanced-learn/install.html')
2828

2929
# This is a tuple to preserve order, so that dependencies are checked
3030
# in some meaningful order (more => less 'core'). We avoid using

0 commit comments

Comments
 (0)