Table 3.
Effect of learning rate on results
Learning rate | Accuracy | Precision | Recall | F1 score | ROC-AUC |
---|---|---|---|---|---|
0.001 | 0.903 | 0.94 | 0.84 | 0.87 | 0.861 |
0.0002 | 0.915 | 0.88 | 0.72 | 0.79 | 0.813 |
ROC-AUC – Receiver operating characteristic-area under the curve