Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
|
|
26 |
def cleaning_features(data):
|
27 |
le = LabelEncoder()
|
28 |
scaler = StandardScaler()
|
29 |
-
encoder = OneHotEncoder(
|
30 |
numeric_cols = ['age', 'educational-num', 'hours-per-week']
|
31 |
columns_to_encode = ['race','marital-status','relationship']
|
32 |
|
|
|
26 |
def cleaning_features(data):
|
27 |
le = LabelEncoder()
|
28 |
scaler = StandardScaler()
|
29 |
+
encoder = OneHotEncoder(sparse=False)
|
30 |
numeric_cols = ['age', 'educational-num', 'hours-per-week']
|
31 |
columns_to_encode = ['race','marital-status','relationship']
|
32 |
|