Update app.py

app.py

@@ -539,9 +539,18 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
             last_lat_scaled, last_lon_scaled, scaler, lat_idx, lon_idx
         )
 
+        # Squeeze arrays to ensure they are 1-dimensional
+        pred_lat = pred_lat.squeeze()
+        pred_lon = pred_lon.squeeze()
+        true_lat = true_lat.squeeze()
+        true_lon = true_lon.squeeze()
+        last_lat_denorm = last_lat_denorm.squeeze()
+        last_lon_denorm = last_lon_denorm.squeeze()
+        mmsi_seq = mmsi_seq.squeeze()
+
         # Calculate the classic evaluation metrics
-        y_true_pairs = np.column_stack((true_lat.flatten(), true_lon.flatten()))
-        y_pred_pairs = np.column_stack((pred_lat.flatten(), pred_lon.flatten()))
+        y_true_pairs = np.column_stack((true_lat, true_lon))
+        y_pred_pairs = np.column_stack((pred_lat, pred_lon))
         classic_metrics = calculate_classic_metrics(y_true=y_true_pairs, y_pred=y_pred_pairs)
         classic_metrics['Inference Time (seconds)'] = inference_time  # Include inference time
 
@@ -559,14 +568,14 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
 
         # Prepare predicted and real positions DataFrame, including error in Km
         predicted_df = pd.DataFrame({
-            'MMSI': mmsi_seq[:len(y_pred)].flatten(),
-            'Last Known Latitude': last_lat_denorm.flatten(),
-            'Last Known Longitude': last_lon_denorm.flatten(),
-            'Predicted Latitude': pred_lat.flatten(),
-            'Predicted Longitude': pred_lon.flatten(),
-            'Real Latitude': true_lat.flatten(),
-            'Real Longitude': true_lon.flatten(),
-            'Error (Km)': error_km  # Add the error in Km as a new column
+            'MMSI': mmsi_seq[:len(y_pred)],
+            'Last Known Latitude': last_lat_denorm,
+            'Last Known Longitude': last_lon_denorm,
+            'Predicted Latitude': pred_lat,
+            'Predicted Longitude': pred_lon,
+            'Real Latitude': true_lat,
+            'Real Longitude': true_lon,
+            'Error (Km)': error_km
         })
 
         # Save predictions as CSV
@@ -580,6 +589,7 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
         logging.error(f"An error occurred: {str(e)}")
         return None, None, None, str(e)
 
+
 # ============================
 # Abnormal Behavior Detection
 # ============================