Update app.py

app.py

@@ -41,6 +41,22 @@ REQUIRED_COLUMN_ORDER = [
 
 # @app.post('/predict')
 # def predict_score(lat, lon):
+
+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
     
 class Location(BaseModel):
     lat: float
@@ -93,7 +109,8 @@ def predict_score(location: Location, request: Request):
 
     # 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)
 
     input_dict_new.update({
         "score":round(float(score), 3),