| seed: 42 | |
| data: | |
| root: data | |
| file: data/ppnn_virtual.npz | |
| format_version: ppnn_nn_aux_emb_v1 | |
| dates: [2020-01-15, 2020-07-15, 2021-01-15, 2021-07-15] | |
| active_station_count: 12 | |
| station_count: 537 | |
| member_count: 50 | |
| variable_count: 18 | |
| lead_hours: 48 | |
| target_variable: t2m | |
| model: | |
| name: PPNN | |
| variant: NN-aux-emb | |
| hidden_size: 32 | |
| paper_hidden_size: 512 | |
| embedding_dim: 2 | |
| continuous_features: 40 | |
| input_features: 42 | |
| sigma_epsilon: 0.0001 | |
| train: | |
| replicas: 2 | |
| paper_replicas: 10 | |
| epochs: 12 | |
| batch_size: 64 | |
| learning_rate: 0.003 | |
| runtime: | |
| device: auto | |
| paths: | |
| checkpoint: result/checkpoints/ppnn.pt | |
| training_metrics: result/training/metrics.json | |
| inference: result/output/predictions.npz | |
| evaluation_dir: result/evaluation | |