preprocesssing

autonlab · Apr 21, 2022 · 548501b · 548501b
1 parent c6b6a72
commit 548501b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/auton_survival/preprocessing.py b/auton_survival/preprocessing.py
@@ -376,3 +376,9 @@ def fit_transform(self, data, cat_feats, num_feats,
       output = pd.get_dummies(output, dummy_na=False, drop_first=True)
 
     return output
+
+def _dataframe_to_array(data):
+  if isinstance(data, (pd.Series, pd.DataFrame)):
+    return data.to_numpy()
+  else:
+    return data