Bug fixed in Gunnar Raetsch datasets loader

daviddiazvico · Sep 15, 2018 · dc64b09 · dc64b09
1 parent 05d2a7c
commit dc64b09
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/skdatasets/gunnar_raetsch.py b/skdatasets/gunnar_raetsch.py
@@ -44,6 +44,7 @@ def load(name, return_X_y=False):
     X, y, train_splits, test_splits = loadmat(filename)[name][0][0]
     if (len(y.shape) == 1) or (np.prod(y.shape[1:]) == 1):
         y = y.ravel()
+    y[y == -1] = 0
     inner_cv = CustomSplit(train_splits[:5] - 1, test_splits[:5] - 1)
     outer_cv = CustomSplit(train_splits - 1, test_splits - 1)
     if return_X_y: