fix na in the dataset

MAIF · Oct 10, 2024 · 876227d · 876227d
1 parent 368c154
commit 876227d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eurybia/core/smartdrift.py b/eurybia/core/smartdrift.py
@@ -273,6 +273,7 @@ def compile(
         dtypes = df_concat[varz].dtypes.map(str)
         cat_features = list(dtypes[dtypes.isin(["object"])].index)
         df_concat[cat_features] = df_concat[cat_features].fillna("NA")
+        df_concat = df_concat.fillna(0)
         self._df_concat = df_concat
 
         train, test = train_test_split(df_concat[varz + [self._datadrift_target]], test_size=0.25, random_state=42)