matsammut commited on
Commit
af74c65
·
verified ·
1 Parent(s): 3012370

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -33,6 +33,8 @@ def cleaning_features(data):
33
 
34
  data['gender'] = le.fit_transform(data['gender'])
35
  data['educational-num'] = le.fit_transform(data['educational-num'])
 
 
36
 
37
  data[numeric_cols] = scaler.fit_transform(data[numeric_cols])
38
 
 
33
 
34
  data['gender'] = le.fit_transform(data['gender'])
35
  data['educational-num'] = le.fit_transform(data['educational-num'])
36
+ data['workclass'] = le.fit_transform(data['workclass'])
37
+ data['occupation'] = le.fit_transform(data['occupation'])
38
 
39
  data[numeric_cols] = scaler.fit_transform(data[numeric_cols])
40