Update utils.py

utils.py

@@ -6,6 +6,7 @@ from scipy.spatial import cKDTree
 df_amenities = pd.read_csv("df_amenities.csv")
 df_banks = pd.read_csv("df_banks.csv")
 
+
 bank_coords = df_banks[['lat','lon']].values
 tree_banks = cKDTree(bank_coords)
 
@@ -13,11 +14,11 @@ amenity_coords = df_amenities[['lat','lon']].values
 tree_amenities = cKDTree(amenity_coords)
 
 DATASET_COLUMNS = [
-    'num_Dining and Drinking', 'num_Community and Government', 'num_Retail',
-       'num_Business and Professional Services', 'num_Landmarks and Outdoors',
-       'num_Arts and Entertainment', 'num_Health and Medicine',
-       'num_Travel and Transportation', 'num_Sports and Recreation',
-       'num_Event'
+    'Dining and Drinking', 'Community and Government', 'Retail',
+       'Business and Professional Services', 'Landmarks and Outdoors',
+       'Arts and Entertainment', 'Health and Medicine',
+       'Travel and Transportation', 'Sports and Recreation',
+       'Event'
 ]
 
 def compute_features(candidate_point, radius=0.005):
@@ -25,6 +26,8 @@ def compute_features(candidate_point, radius=0.005):
 
     # Banks
     bank_idxs = tree_banks.query_ball_point([lat, lon], r=radius)
+
+    print("[BANK]", bank_idxs)
     
     n_banks = len(bank_idxs)
     if n_banks > 0: