Streamflow-LSTM / conf /config.yaml
zhangrenchao's picture
Publish Streamflow-LSTM reproduction
186a48a verified
Raw History Blame Contribute Delete
1.19 kB
{
"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"
}
}