Table 1 Parameter setup for different machine learning methods
From: Blockchained supply chain management based on IoT tracking and machine learning
Parameter | GRU | RNN |
|---|---|---|
Network structure | Data normalization multi-head attention—GRU (three head) | Data normalization three layers RNN |
Learning rate(lr) | 0.01 | 0.01 |
lr scheduler | MultiStepLR | MultiStepLR |
Optimizer | SGD | SGD |
Loss function | RMSE | RMSE |
Batch size | 56 | 56 |
Epoch | 80 | 80 |
Randomseed | 0 | 0 |