From f4b28ee6fb4a6a58ae4dc6e59950cf346f4ce895 Mon Sep 17 00:00:00 2001 From: Lars Bogner Date: Tue, 16 Dec 2025 18:56:49 +0100 Subject: [PATCH] Edit dimensions --- Code/python/notebooks/predict.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/Code/python/notebooks/predict.py b/Code/python/notebooks/predict.py index 0edc8ce..1981e82 100644 --- a/Code/python/notebooks/predict.py +++ b/Code/python/notebooks/predict.py @@ -65,8 +65,14 @@ with torch.no_grad(): X_c = X_c.to(DEVICE) y_pred, _ = model(X_t, X_f, X_c, warm // step, pred // step) lat_pred, lon_pred, alt_pred = denorm_coords(y_pred[...,0], y_pred[...,1], y_pred[...,2]) + lat_pred = lat_pred.flatten() + lon_pred = lon_pred.flatten() + alt_pred = alt_pred.flatten() lat_true, lon_true, alt_true = denorm_coords(y[...,0], y[...,1], y[...,2]) - results.append(torch.stack([lat_true, lon_true, alt_true, lat_pred, lon_pred, alt_pred], dim=1).cpu()) + lat_true = lat_true.flatten() + lon_true = lon_true.flatten() + alt_true = alt_true.flatten() + results.append(torch.cat([lat_true, lon_true, alt_true, lat_pred, lon_pred, alt_pred], dim=-1).cpu()) np_y_all = torch.cat(results, dim=0).numpy() np.savetxt(