Table 2.

Neural network architectures and hyperparameters used

ParameterNN-1DNN-2DNN-3D
Model typeSequentialSequentialSequential
Input features223
Layer 1 units323232
Layer 1 activationtanhtanhtanh
Layer 2 units8816
Layer 2 activationtanhtanhtanh
Layer 3 units448
Layer 3 activationtanhtanhtanh
Layer 4 units4
Layer 4 activationtanh
Output layer units111
Output layer activationReLUReLUReLU
Loss FunctionMSEMSEMSE
OptimizerAdamAdamAdam
MetricMAPE, RMSEMAPE, RMSEMAPE, RMSE
Epochs2,0002,0002,000
Batch size1,064
(256 in k-fold)
1,064
(256 in k-fold)
1,064
(256 in k-fold)
ShufflingEnabledEnabledEnabled
Validation split0.40.40.4
k-fold c.v., RMSE training-validation0.056–0.0560.12–0.141.29–1.31
Accuracy97%96%85%
Source: By authors

or Create an Account

Close subscription notice
Close access options