Ozone_ML/data.yaml at main · wkeenan21/Ozone_ML · GitHub

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
46
47
48
49
50
51
52
53
54
55
56
57
batch_size: 256
clip_gradient_norm: 1
clip_targets_to_zero:
- o3
data_dir: D:\Will_Git\Ozone_ML\Year2\Merged_Data\nh
dataset: ozone_dataset
device: None
dynamic_inputs:
- o3_6day
- u10
- v10
- r2
- sp
- t2m
- dswrf
- MAXUVV
- MAXDVV
epochs: 25
experiment_name: test_run
head: regression
hidden_size: 20
initial_forget_bias: 3
learning_rate:
  0: 1e-2
  30: 5e-3
  40: 1e-3
log_interval: 5
log_n_figures: 1
log_tensorboard: true
loss: RMSE
metrics:
- NSE
- RMSE
- Pearson-r
model: cudalstm
num_workers: 8
optimizer: Adam
output_activation: linear
output_dropout: 0.4
predict_last_n:
  1H: 1
save_weights_every: 1
seq_length: 48
target_variables:
- o3
test_basin_file: D:\Will_Git\Ozone_ML\Year2\scripts_dir\neuralHydro\trainingSites.txt
test_end_date: 06/08/2022
test_start_date: 07/03/2022
train_basin_file: D:\Will_Git\Ozone_ML\Year2\scripts_dir\neuralHydro\trainingSites.txt
train_end_date: 06/03/2022
train_start_date: 04/05/2021
use_frequencies: 1H
validate_every: 5
validate_n_random_basins: 1
validation_basin_file: D:\Will_Git\Ozone_ML\Year2\scripts_dir\neuralHydro\trainingSites.txt
validation_end_date: 06/08/2022
validation_start_date: 07/03/2021