yusufenes commited on
Commit
430f47b
·
verified ·
1 Parent(s): 574a624

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ from sklearn.ensemble import HistGradientBoostingRegressor
18
  hgbr = HistGradientBoostingRegressor()
19
  num_features = X.select_dtypes(include=['int64']).columns
20
  cat_features = X.select_dtypes(include='object').columns
21
- transformer = ColumnTransformer(transformers=[('num', StandardScaler(), num_features),('cat', OneHotEncoder(sparse=0), cat_features)])
22
  pipe = Pipeline(steps=[('preprocessor', transformer), ('model', hgbr)])
23
 
24
  pipe.fit(X_train, y_train)
 
18
  hgbr = HistGradientBoostingRegressor()
19
  num_features = X.select_dtypes(include=['int64']).columns
20
  cat_features = X.select_dtypes(include='object').columns
21
+ transformer = ColumnTransformer(transformers=[('num', StandardScaler(), num_features),('cat', OneHotEncoder(sparse_output=False), cat_features)])
22
  pipe = Pipeline(steps=[('preprocessor', transformer), ('model', hgbr)])
23
 
24
  pipe.fit(X_train, y_train)