Skip to content

Commit 876227d

Browse files
fix na in the dataset
1 parent 368c154 commit 876227d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

eurybia/core/smartdrift.py

+1
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,7 @@ def compile(
273273
dtypes = df_concat[varz].dtypes.map(str)
274274
cat_features = list(dtypes[dtypes.isin(["object"])].index)
275275
df_concat[cat_features] = df_concat[cat_features].fillna("NA")
276+
df_concat = df_concat.fillna(0)
276277
self._df_concat = df_concat
277278

278279
train, test = train_test_split(df_concat[varz + [self._datadrift_target]], test_size=0.25, random_state=42)

0 commit comments

Comments
 (0)