Fix test_proper_boundaries

Author: vnk8071

README.md

@@ -126,3 +126,8 @@ mlflow run . -P steps=test_regression_model
 2023-08-21 20:52:32,341 Score: 0.5640242000942114
 2023-08-21 20:52:32,341 MAE: 33.850181065122136
 ```
+
+## 5. Public W&B project
+```bash
+Link: https://wandb.ai/nguyenkhoi8071/nyc_airbnb/overview?workspace=user-nguyenkhoi8071
+```

src/basic_cleaning/run.py

@@ -30,8 +30,11 @@ def go(args):
     # Drop outliers
     logger.info("Dropping outliers")
     logger.info(f"Number of rows before dropping outliers: {df.shape[0]}")
-    idx = df['price'].between(args.min_price, args.max_price)
-    df = df[idx].reset_index(drop=True)
+    idx_price = df['price'].between(args.min_price, args.max_price)
+    df = df[idx_price].reset_index(drop=True)
+    idx = df['longitude'].between(-74.25, -
+                                  73.50) & df['latitude'].between(40.5, 41.2)
+    df = df[idx].copy()
     logger.info(f"Number of rows after dropping outliers: {df.shape[0]}")
 
     # Convert last_review to datetime