Fix bug that removes all information of rows with remove in a column

lukassnoek · Sep 23, 2016 · 1dcb053 · 1dcb053
1 parent 2a24ecf
commit 1dcb053
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skbold/utils/crossval_splitter.py b/skbold/utils/crossval_splitter.py
@@ -15,11 +15,11 @@ def __init__(self, file_path, train_size, test_size=0, categorical={}, continuou
 
         if len(categorical)>0:
             for cat in categorical.keys():
-                data.loc[data[cat] == ignore] = np.nan  # ignore values, such as 9999
+                data.loc[data[cat] == ignore, cat] = np.nan  # ignore values, such as 9999
 
         if len(continuous)>0:
             for cont in continuous:
-                data.loc[data[cont] == ignore] = np.nan  # ignore values, such as 9999
+                data.loc[data[cont] == ignore, cont] = np.nan  # ignore values, such as 9999
 
         self.data = data
         self.train_size = train_size