Table 6.
Comparison of the proposed 3D CNN trained by different batch size and initial learning rate.
| Batch size | Learning rate | MAE (days) | RMSE (days) | MRE (%) | R | R2 |
|---|---|---|---|---|---|---|
| 16 | 0.0000008 | 72.3 | 96.2 | 8.3 | 0.985 | 0.971 |
| 32 | 0.0000008 | 74.1 | 97.4 | 9.2 | 0.985 | 0.971 |
| 64 | 0.0000008 | 67.6 | 96.1 | 8.2 | 0.985 | 0.971 |
| 64 | 0.0000012 | 69.9 | 97.1 | 8.6 | 0.985 | 0.971 |
| 64 | 0.0000004 | 75.0 | 104.8 | 9.3 | 0.983 | 0.966 |