diff --git a/config.yml b/config.yml index ed48096554eb238ed0e939cda9a595372819684c..09e83255cb3ff718960ccb281dbedd49d780e5ac 100644 --- a/config.yml +++ b/config.yml @@ -1,6 +1,6 @@ # Dataset Configuration Dataset: - num_days: 1095 # Test with sequence of 1 day + num_days: 73 # Test with sequence of 1 day - should be the same as in Test - batch_size: 64 num_workers: 7 valid_ratio: 0.2 @@ -13,7 +13,8 @@ Dataset: _ENCODING_ENDIAN: "<" # Data Transformation -ApproximativeStats: True +ApproximativeStats: False + ApproximativeMean: "torch.tensor([ 4.2457e+01, 7.4651e+00, 1.6738e+02, 1.3576e+09, 2.3628e+00, 4.6839e+01, 2.3855e-01, 3.6535e+00, 1.9776e+00, 2.2628e+02, 8.1003e+00, 1.8691e-01, 3.8384e+01, 2.6626e+00, 1.4315e+01, @@ -39,7 +40,7 @@ Training: #Model selection Model: - Name: BidirectionalLSTM + Name: RNN #Model parameters selection LinearRegression: @@ -52,7 +53,11 @@ BidirectionalLSTM: HiddenSize: 70 NumLayers: 4 Initialization: None - + +RNN: + HiddenSize: 35 + NumLayers: 4 + Initialization: None #Name of directory containing logs LogDir: ./logs/