Ukemelem commited on
Commit
b7375fa
1 Parent(s): 513483f

fixed error in app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  from tensorflow.keras.models import load_model
 
2
  import pandas as pd
3
 
4
  saved_model = load_model('churn_model2.h5')
 
5
 
6
  def churn_prediction(CreditScore,Gender,Age,Tenure,Balance,NumOfProducts,HasCrCard,IsActiveMember,EstimatedSalary,Location):
7
  Geography_France,Geography_Germany,Geography_Spain=0,0,0
@@ -21,7 +23,7 @@ def churn_prediction(CreditScore,Gender,Age,Tenure,Balance,NumOfProducts,HasCrCa
21
 
22
  })
23
  cols_to_scale = ["Credit score",'Age','Tenure','Balance','Number of products','Estimated Salary']
24
- df[cols_to_scale] = scaler.transform(df[cols_to_scale])
25
  pred=saved_model.predict(df)
26
  pred = pred[0][0]
27
  churn_prob=str(round(pred,2))
 
1
  from tensorflow.keras.models import load_model
2
+ from sklearn.preprocessing import MinMaxScaler
3
  import pandas as pd
4
 
5
  saved_model = load_model('churn_model2.h5')
6
+ scaler = MinMaxScaler()
7
 
8
  def churn_prediction(CreditScore,Gender,Age,Tenure,Balance,NumOfProducts,HasCrCard,IsActiveMember,EstimatedSalary,Location):
9
  Geography_France,Geography_Germany,Geography_Spain=0,0,0
 
23
 
24
  })
25
  cols_to_scale = ["Credit score",'Age','Tenure','Balance','Number of products','Estimated Salary']
26
+ df[cols_to_scale] = scaler.fit_transform(df[cols_to_scale])
27
  pred=saved_model.predict(df)
28
  pred = pred[0][0]
29
  churn_prob=str(round(pred,2))