File size: 1,194 Bytes
186a48a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
{
  "seed": 2026,
  "data": {
    "format_version": "streamflow-lstm-v1",
    "path": "data/streamflow_fake.npz",
    "gauges": ["BNDN5", "ARWN8", "TCCC1", "CARO2", "ESSC2", "NFDC1", "LABW4", "CLNK1", "TRAC2", "NFSW4"],
    "input_variables": 23,
    "history_steps": 28,
    "step_hours": 6,
    "forecast_steps": 40,
    "train_samples": 48,
    "validation_samples": 12,
    "forecast_cases": 12
  },
  "model": {
    "hidden_size": 16,
    "num_layers": 3,
    "dropout": 0.1
  },
  "paper_model": {
    "hidden_size": 50,
    "num_layers": 3,
    "activations": ["relu", "relu", "tanh"],
    "ensemble_members_per_gauge": 100,
    "best_members": 5,
    "epochs": 10
  },
  "training": {
    "ensemble_members_per_gauge": 2,
    "best_members": 2,
    "epochs": 10,
    "batch_size": 16,
    "learning_rate": 0.001,
    "loss": "MSE"
  },
  "runtime": {"device": "auto", "num_threads": 2},
  "paths": {
    "checkpoint": "result/checkpoints/streamflow_lstm.pt",
    "training_metrics": "result/training/metrics.json",
    "predictions": "result/output/predictions.npz",
    "evaluation_metrics": "result/evaluation/metrics.json",
    "comparison": "result/evaluation/comparison.png"
  }
}