Update app.py

app.py

@@ -28,6 +28,22 @@ REQUIRED_COLUMN_ORDER = [
     'num_viewpoint', 'num_jewelry_store'
 ]
 
+def atm_score(num_atm, num_prediction_atm, k=7, pct=0.15):
+    # num_atm = np.asarray(num_atm, dtype=float)
+    # num_prediction_atm = np.asarray(num_prediction_atm, dtype=float)
+
+    # Avoid division by zero
+    eps = 1e-8
+    # num_atm = num_atm*(1 + pct)
+    num_prediction_atm = num_prediction_atm*(1 + pct)
+
+    # delta = (num_atm - num_prediction_atm) / (num_prediction_atm + eps)
+    delta = (num_prediction_atm - num_atm) / (num_atm + eps)
+
+    score = 100.0 / (1.0 + np.exp(-k * (delta)))
+    # score = 100 * np.exp( -k*delta ** 2 /( 300*(0.085)**2))
+    return score
+
 
 def predict_score(lat, lon, api_key):
     # Convert input to tensor
@@ -73,7 +89,8 @@ def predict_score(lat, lon, api_key):
 
     # score = np.sigmoid(mu_pred2 - num_banks + 0.1) * 100
 
-    score =  100 / (1 + np.exp(num_banks - mu_pred))
+    # score =  100 / (1 + np.exp(num_banks - mu_pred))
+    score = atm_score(num_banks, mu_pred)
 
     # You can apply any post-processing here
     return (