Table 2.
Hyper-parameter values for training.
| Hyper-parameters | Value |
|---|---|
| Epochs | 10 |
| Learning rate | 0.001 |
| Batch size | 16 |
| Optimizer | Adam |
| beta_1 | 0.9 |
| beta_2 | 0.999 |
| Loss function | MSE |
Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.