Table 3.
Parameter | Value |
---|---|
Initial learning rate | |
Maximum epochs | 50 |
Batch size | 32 |
Early stopping | 10 epochs |
Loss function | Cross-entropy weighted for class frequency |
Optimizer | AdamW |
Learning rate scheduler | Reduce learning rate on plateau |
Parameter | Value |
---|---|
Initial learning rate | |
Maximum epochs | 50 |
Batch size | 32 |
Early stopping | 10 epochs |
Loss function | Cross-entropy weighted for class frequency |
Optimizer | AdamW |
Learning rate scheduler | Reduce learning rate on plateau |