Update README.md

README.md

@@ -51,7 +51,7 @@ with torch.no_grad():  # Disable gradient calculations during inference
 all_preds = np.array(all_preds)
 all_actuals = np.array(all_actuals)
 
-all_preds_denorm = all_preds * np.array([lat_std, lon_std]) + np.array([lat_mean, lon_mean])  # Assuming lat_std, etc., are lists or NumPy arrays
+all_preds_denorm = all_preds * np.array([lat_std, lon_std]) + np.array([lat_mean, lon_mean])
 all_actuals_denorm = all_actuals * np.array([lat_std, lon_std]) + np.array([lat_mean, lon_mean]) 
 
 squared_errors = []