Skip to content

Commit bb49e8d

Browse files
author
dcolinmorgan
committed
sweet spot hack
1 parent f893591 commit bb49e8d

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

cu_cat/_table_vectorizer.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -631,9 +631,8 @@ def fit_transform(self, X, y=None):
631631
if self.verbose:
632632
print(f"[TableVectorizer] Assigned transformers: {self.transformers}")
633633
if 'cudf' in self.Xt_ and 'cudf' not in str(getmodule(X)):
634-
X=cudf.from_pandas(X)#,nan_as_null=False)
635-
# print(str(getmodule(y)))
636-
# y=cudf.from_pandas(y); should already be in cudf since not manipulated earlier
634+
X=cudf.from_pandas(X)#,nan_as_null=True) ### see how flag acts
635+
X = X.fillna(0)
637636
X_enc = super().fit_transform(X, y)
638637

639638
X_enc = cudf.DataFrame(X_enc) #from cuml.arraydata

0 commit comments

Comments
 (0)