Spaces:
Runtime error
Runtime error
fixed error in app.py
Browse files
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.
|
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))
|