From 1dcb05332e2c8a0c718801a00a00a456d1843ba0 Mon Sep 17 00:00:00 2001 From: Steef Date: Fri, 23 Sep 2016 13:44:30 +0200 Subject: [PATCH] Fix bug that removes all information of rows with remove in a column --- skbold/utils/crossval_splitter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skbold/utils/crossval_splitter.py b/skbold/utils/crossval_splitter.py index 5b61cd3..2f89fe6 100644 --- 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