lars/aiRtrafficNN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
Lars Bogner
2025-12-16 18:53:50 +01:00
parent 87233e0f56
commit 5895bbaed3
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Code/python/notebooks/predict.py
+3 -5
View File
@@ -66,11 +66,9 @@ with torch.no_grad():
y_pred, _ = model(X_t, X_f, X_c, warm // step, pred // step) 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, lon_pred, alt_pred = denorm_coords(y_pred[...,0], y_pred[...,1], y_pred[...,2])
lat_true, lon_true, alt_true = denorm_coords(y[...,0], y[...,1], y[...,2]) lat_true, lon_true, alt_true = denorm_coords(y[...,0], y[...,1], y[...,2])
results.append((torch.stack([lat_true, lon_true, alt_true], dim=-1).cpu(), torch.stack([lat_pred, lon_pred, alt_pred], dim=-1).cpu())) results.append(torch.stack([lat_true, lon_true, alt_true, lat_pred, lon_pred, alt_pred], dim=1).cpu())
np_y_all = np.concatenate([
np.concatenate([t.numpy(), p.numpy()], axis=-1) np_y_all = torch.cat(results, dim=0).numpy()
for t, p in results
], axis=0)
np.savetxt( np.savetxt(
f"predictions_{base_name}_wu{warm}_ps{pred}_aw{altitude_weight}.csv", f"predictions_{base_name}_wu{warm}_ps{pred}_aw{altitude_weight}.csv",
np_y_all, np_y_all,