Research Article
Prediction of Visual Acuity after anti-VEGF Therapy in Diabetic Macular Edema by Machine Learning
Table 2
Accuracy of visual acuity predictions.
| Algorithm learner | MAE | MSE | All features | Selected features | All features | Selected features |
| Linear regression (LR) | 0.153 | 0.149 | 0.048 | 0.046 | SVM | 0.272 | 0.359 | 0.140 | 0.188 | K neighbors regressor | 0.222 | 0.195 | 0.076 | 0.060 | Random forest regressor (RF) | 0.168 | 0.153 | 0.050 | 0.042 | Ridge regressor | 0.183 | 0.159 | 0.058 | 0.046 | LR + RF | 0.153 | 0.137 | 0.045 | 0.033 |
|
|
MAE, mean absolute error; MSE, mean square error; accuracy (VA in logMAR) of VA prediction at 1 month after anti-VEGF compared with the ground truth.
|