Spaces:

Synav
/

Explainable-Acute-Leukemia-Mortality-Predictor

Running

Synav commited on Jan 23

Commit

21da4e4

verified ·

1 Parent(s): 0954c79

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ def build_pipeline(
     cat_pipe = Pipeline([
         ("imputer", SimpleImputer(strategy="most_frequent")),
-        ("onehot", OneHotEncoder(handle_unknown="ignore", sparse_output=True))
     ])
     preprocessor = ColumnTransformer(

     cat_pipe = Pipeline([
         ("imputer", SimpleImputer(strategy="most_frequent")),
+        ("onehot", OneHotEncoder(handle_unknown="ignore", sparse_output=True, drop="if_binary"))
     ])
     preprocessor = ColumnTransformer(