Application of Stacking Ensemble Learning in Clinical Fitting of Orthokeratology Lens for Myopia Correction
Aiming at the problems of a large difficulty coefficient and tedious process in the clinical fitting of the orthokeratology(OK)lens,a stacking ensemble learning model is proposed to predict the parameters of the OK lens and realize its intelligent fitting.The feature set that is most suitable for the target variables is constructed by feature derivation based on F-test and feature selection under the variance-improved Boruta algorithm.A stacking ensemble learning prediction model is studied.The model uses random forest(RF),gradient boosting decision tree(GBDT)and support vector regression(SVR)as the first layer basic learners and linear regression(LR)as the second layer meta-learner.The experimental results show that the prediction indexes of the model are highly consistent with the clinical diagnosis results,which verifies that the model can be used as an effective auxiliary diagnosis method.