matsammut commited on
Commit
7702d79
·
verified ·
1 Parent(s): ce3be16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,6 +21,8 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
21
  fixed_features = cleaning_features(df)
22
  # with open('ann_model.pkl', 'rb') as ann_model_file:
23
  # ann_model = pickle.load(ann_model_file)
 
 
24
  ann_model = load_model('ann_model.h5')
25
  prediction = ann_model.predict(fixed_features)
26
  # prediction = 1
 
21
  fixed_features = cleaning_features(df)
22
  # with open('ann_model.pkl', 'rb') as ann_model_file:
23
  # ann_model = pickle.load(ann_model_file)
24
+ scaler = StandardScaler()
25
+ X = scaler.fit_transform(fixed_features)
26
  ann_model = load_model('ann_model.h5')
27
  prediction = ann_model.predict(fixed_features)
28
  # prediction = 1