Research Article

Research on Disease Prediction Method Based on R-Lookahead-LSTM

Algorithm 2

R-Lookahead-LSTM algorithm.
R-Lookahead-LSTM algorithm
 Step 1: process the data, analyze the correlation of the data, and use the ensemble learning algorithm random forest algorithm to perform feature selection on the data, so as to further determine the feature vector required to build the model
 Step 2: divide the above-processed data set into a training set and test set according to the ratio of 7:3
 Step 3: determine the structure of the LSTM model for the data samples of the training set and determine the number of network layers and initialization parameters through experimental tests
 Step 4: train the model and use the R-Lookahead optimization algorithm to optimize the loss function of the LSTM model
 Step 5: use the test set data samples as the input data of the model to test the prediction effect of the R-Lookahead-LSTM disease risk prediction model
 Step 6: use multiple indicators such as accuracy rate, recall rate, F1-score, specificity, and MCC value to evaluate the prediction effect of the model