fix improper line drop

This commit is contained in:
Andy Eschbacher 2016-06-27 09:58:42 -04:00
parent a5e4ae99ce
commit a2b3733a1e

View File

@ -57,8 +57,7 @@ def get_data(variable, feature_columns, query):
# put arrays into an n x m array of arrays # put arrays into an n x m array of arrays
features = np.column_stack([np.array(data[0][col], dtype=float) for col in feature_columns]) features = np.column_stack([np.array(data[0][col], dtype=float) for col in feature_columns])
return replace_nan_with_mean(target), return replace_nan_with_mean(target), replace_nan_with_mean(features)
replace_nan_with_mean(features)
# High level interface # High level interface
# -------------------- # --------------------