Forecast of the COVID-19 Epidemic Based on RF-BOA-LightGBM

Abstract

In this paper, we utilize the Internet big data tool, namely Baidu Index, to predict the development trend of the new coronavirus pneumonia epidemic to obtain further data. By selecting appropriate keywords, we can collect the data of COVID-19 cases in China between 1 January 2020 and 1 April 2020. After preprocessing the data set, the optimal sub-data set can be obtained by using random forest feature selection method. The optimization results of the seven hyperparameters of the LightGBM model by grid search, random search and Bayesian optimization algorithms are compared. The experimental results show that applying the data set obtained from the Baidu Index to the Bayesian-optimized LightGBM model can better predict the growth of the number of patients with new coronary pneumonias, and also help people to make accurate judgments to the development trend of the new coronary pneumonia.

1. Introduction

During the outbreak of infectious diseases, social media is usually the most active platform for the exchange of information on infectious disease, and the information released is often of good real-time. Using Internet information to predict the epidemic situation of infectious diseases is one of the current research hotspots. L. Lu et al. used Baidu index and micro-index to conduct a comparative study on influenza surveillance in China [1]. J. H. Lu, School of Public Health, Sun Yat-sen University, and others studied the use of Internet search queries or social media data to monitor the temporal and spatial trends of the Avian Influenza (H7N9) in China, and the results show that the number of H7N9 cases is positively correlated with Baidu Index and Weibo Index search results in space and time [2]. J. X. Feng of the University of South Georgia and others studied the impact of Chinese social networks on the Middle East Respiratory Syndrome Coronavirus and Avian Influenza [3]. Mutual relations prove the effectiveness of using social media to predict infectious diseases. H. G. Gu et al. collected data on cases of H7N9 avian influenza in the Chinese urban population through the Internet, as well as geographic and meteorological data during the same period, and established a disease risk early warning model for human infection with H7N9 avian influenza, which can identify the high risk areas of avian influenza outbreaks and issue an early warning [4]. However, in these studies, most of the search process of network data adopts manual empirical methods to select keywords for search, and the choice of keywords often has a greater impact on search results.

At present, the focus of the world’s attention is mainly on the changes in the epidemic situation of the new type of coronary pneumonia. During the four months after the outbreak of the new type of coronavirus in Wuhan, Hubei in December 2019, the epidemic information was widely disseminated on social media such as Baidu, Sina, 360, Sogou, WeChat and QQ. Google, Weibo, Zhihu, Dingxiangyuan, Twitter, Facebook, etc. also released a lot of information about the new coronavirus epidemic, especially through the Google platform to spread to the world. On 31 March 2019, Google launched a project called “COVID-19 Public Datasets” to provide a public database related to the epidemic and open it to the public for free, which means that people can freely access and analyze relevant data and information [5]. How to use this information to predict the spread of the new type of coronary pneumonia in time is an urgent research topic. Currently, X. M. Zhao and others have proposed to use big data retrospective technology to study the spreading trend and epidemic control of the new coronary pneumonia [6]. B. McCall et al. used artificial intelligence methods to predict the new type of coronary pneumonia, thereby protecting medical staff and controlling the spread of the epidemic [7]. These studies are still in the preliminary stage, and the use of network data and prediction of the new coronary pneumonia are not yet ideal.

In this article, we consider that the amount of data indexed by Baidu is large enough for us to use. Based on this, we use the first feature in the search index, namely Baidu index [8], to study the prediction of the epidemic of new coronary pneumonia. We collected data on COVID-19 cases in China from 1 January 2020 to 1 April 2020, and used the random forest feature selection method to select the optimal sub-data set, and used grid search, random search and the Bayesian optimization algorithm optimizes the 7 hyperparameters of the LightGBM (light gradient boosting machine) model. The results show that the application of the data set obtained from the Baidu index to the Bayesian-optimized LightGBM model can better predict the growth of the number of patients with new coronary pneumonia.

This paper is organized as follows. In Section 2, we introduce the data set and analysis method used in detail. Baidu index search and actual case results are compared in time and space, and the impact of keywords and selected index in Baidu index search on the results is analyzed. Model structure, data set preprocessing methods, tuning algorithm, etc. are also introduced in detail. In Section 3, the experimental results are showed and related discussions are presented. Finally, the conclusion is drawn in Section 4.

2. Materials and Methods

2.1. COVID-19 Dataset

In order to standardize prevention and treatment, on 11 February 2020, the World Health Organization named the pneumonia caused by the new coronavirus as “COVID-19” (Corona Virus Disease 2019). In this study, we first obtain the data of COVID-19 cases that occurred in China from 1 January 2020 to 1 April 2020 by searching the COVID-19 Public Datasets on the Google platform, mainly including diagnosis number and death toll, and use them as actual data. These data are released by the Centers for Disease Control (CDC), so we identify these data as CDC data , namely the CDC-Diagnosis and CDC-Death toll mentioned in this paper. Then, we can collect keywords related to COVID-19 through commonly used social networking sites, such as Baidu, Sina, 360, Sogou, WeChat, QQ, Google, Weibo, Zhihu, Dingxiangyuan, Twitter, Facebook, etc., And form a keyword library. Then use the Baidu index platform (http://index.baidu.com, (accessed on 1 April 2020)) to retrieve relevant keywords, and use the statistics of the average daily search volume of relevant Chinese keywords as social network mining data for prediction. In this article, this part of the data is identified as Baidu index data.

By searching for the name and clinical symptoms of new coronavirus pneumonia on social networking sites, we can get the following keywords: new coronavirus, fever, dry cough, fatigue, dyspnea and cough. Using the Baidu index platform to retrieve the above keywords, we can get the average daily search volume of each keyword from 1 January 2020 to 1 April 2020, that is, Baidu index data.

Table 1. Partial data from CDC and Baidu Index search.

		Source	CDC- Diagnosis	Baidu- New Coronavirus	Baidu- Fever	Baidu- Dry Cough	Baidu- Fatigue	Baidu- Dyspnea	Baidu- Cough	CDC- Death Toll
	Data
Date
1 January 2020			0	0	4001	1100	256	481	5885	0
2 January 2020			0	0	4323	1206	278	602	6448	0
3 January 2020			1	0	4212	1173	262	654	6392	0
4 January 2020			0	0	4309	1109	270	621	6570	0
5 January 2020			5	0	4327	1118	271	591	6564	0
6 January 2020			0	0	4324	1226	310	693	6404	0
7 January 2020			0	0	3920	1175	288	633	5875	0
8 January 2020			0	0	3803	1124	272	622	5354	0
9 January 2020			0	8812	3693	1131	270	579	5182	0
10 January 2020			0	2032	3700	1095	263	535	5022	0
11 January 2020			0	2879	3478	1083	237	498	5033	1
12 January 2020			0	1445	3364	1067	252	474	5011	1
13 January 2020			0	1515	3573	1118	278	494	4418	1
14 January 2020			0	4846	3479	1133	266	528	4359	1
15 January 2020			0	4191	3241	1097	245	512	4355	2

Note: CDC = Centers of Disease Control.

shows part of the data of the CDC data set and the Baidu index data set. See Appendix A for all the data.

2.1.1. Time and Space Comparative Analysis of Baidu Index Search and Actual Cases

Based on the data obtained during the data collection phase, we have drawn the trend graph of CDC data and Baidu Index data over time, as shown in Figure 1. From Figure 1a–g, it can be seen that the keyword “dry cough” is the most commonly used keyword when Chinese netizens search for symptoms of new coronavirus pneumonia, followed by fever, dyspnea, and fatigue. We can see that in the Baidu index method, the keywords “new coronavirus” and “dry cough” are the best choices. The extracted data has the best spatio-temporal positive correlation with the actual number of cases. Through website search, we can find that these two keywords mainly appear in the columns of Baidu Baike and Baidu Health Pharmacopoeia. Therefore, it is recommended to search these two columns first when choosing keywords in the future. On the other hand, it can also be seen that the Baidu index method is used to predict the change trend of the new coronavirus pneumonia. If the keywords are not selected properly, not only will the accuracy of the prediction be low, but sometimes it may even make it impossible to predict in advance.

In addition, we can see that the CDC diagnosis number and Baidu index data have peak times, so we can compare the correlation between the Baidu index data and the CDC-Diagnosis number from the perspective of the first peak generation time and the time difference, which are shown in

Table 2. Comparison of peak time between Baidu Index data based on different keywords and CDC new diagnostic data.

Category	First Peak Time	Time Difference (Days)
CDC-Diagnostic	12 February 2020	-
Baidu-New coronavirus	25 January 2020	+18
Baidu-Fever	26 January 2020	+17
Baidu-Dry cough	23 January 2020	+20
Baidu-Fatigue	25 January 2020	+18
Baidu-Dyspnea	25 January 2020	+18
Baidu-Cough	25 January 2020	+18
Arithmetic mean	-	+18

Note: + indicates the number of days in advance, - indicates the number of days later. CDC = Centers of Disease Control.

. From the comparative analysis of Figure 1 and Table 2, we can draw the following conclusions. The actual number of new coronavirus pneumonia cases in China reached its highest value on 12 February 2020, which was 15,152, while the Baidu Index data all reach their peak before this date, and the average value of the first peak time difference between the Baidu Index data based on the six keywords and the newly diagnosed CDC is 18 days. This is mainly because during the outbreak of the COVID-19, people like to discuss the it on social media networks. The information released on the new crown epidemic is often of good real-time. The CDC data collection comes from the national infectious disease surveillance system, where the pneumonia often requires a longer diagnosis process from onset to diagnosis, usually 7–14 days.

2.1.2. The Influence of the Selected Index on the Result

In order to explore the impact of the selected index on the results, we first need to check the distribution of the number of new coronavirus confirmed in the data, as shown in Figure 2. It can be seen from the figure that the overall distribution of the target variable deviates from the normal distribution and needs to be adjusted later. The skewness and kurtosis are calculated again, and the calculation results are 10.72 and 140.84, respectively. It can basically be determined that the skewness of the data in this paper is relatively large and needs to be adjusted.

Figure 3 shows the Q-Q graph of the COVID-19 data set. Judge whether the data conforms to the normal distribution by comparing whether the quantiles of the data and the normal distribution are equal. The red line represents the normal distribution, and the blue line represents the sample data. The closer the blue and red reference lines are, the more in line with the expected distribution. From the distribution of data, the data presents a normal state. It is further verified that the data distribution has a large skewness, and further data conversion is needed to make it conform to the normal distribution.

Figure 4 shows the relationship between Diagnosis Numbers and other attributes. It can be seen from the figure that the attributes in the data set are basically positively correlated with the attributes of Diagnosis Numbers. Figure 5 shows the relationship between all attributes, which can be represented by a heat map. The heat map uses different colors to intuitively show the relationship between different attributes, which is a very simple way of data interpretation. The values in the figure are calculated using Pearson’s correlation coefficient. The calculation formula of Pearson’s correlation coefficient is

$$r(X,Y)=\frac{Cov(X,Y)}{\sqrt{Var\left[X\right]Var\left[Y\right]}}.$$

(1)

It can be seen from the heat map that the attribute of month is negatively correlated with Diagnosis Numbers. It can be seen from the above analysis that the collected data set has a certain influence on Diagnosis Numbers and can be used for the numerical prediction of Diagnosis Numbers.

2.2. RF-BOA-LightGBM

As a new cutting-edge technology, predictive models based on machine learning have been widely used in various fields of medicine. For example, Y. D. Zhang et al. proposed a new attention network model, namely ANC (attention network for COVID-19) model, which can diagnose COVID-19 more effectively and accurately [9]. X. Zhang et al. enhanced the deep learning network AlexNet to achieve a more effective classification of new coronary pneumonia [10]. Here, we consider using the RF-BOA-LightGBM (random forest-Bayesian optimization algorithm-light gradient boosting machine) model to predict the development trend of the COVID-19.

2.2.1. Model Structure

Figure 6 shows the model structure used in this article. After collecting the data, you need to perform a simple processing on the data, so that this model can “learn” the data. Then build the LightGBM model for training, but due to the many parameters of LightGBM, the effect of using the default parameters to train the data set in this article is not necessarily good, so three hyperparameter tuning algorithms are introduced here to adjust the model parameters of LightGBM Perform tuning. After finding a combination of model parameters suitable for the data set in this article, the training prediction is carried out.

2.2.2. Dataset Preprocessing

In order to enable the model to fully learn the data obtained from the Baidu Index COVID-19 vaccine, this article first made great efforts to preprocess the data. It can be seen from the foregoing that the distribution of the data in this paper presents a similar normal distribution. Therefore, this article first performs logarithmic transformation on the data to make the data satisfy the normal distribution. The data conversion formula is

$$y={log}_c\left(1+{\lambda }_x\right).$$

(2)

Then, deal with the missing data in the data set and delete the samples with missing values (there are not many samples with missing values, which has little effect on the results). Subsequently, the date is divided into three attributes: year, month, and day, and the year attribute is deleted (the year attribute is a fixed value and has little effect on the result), which avoids the problem that the model cannot directly process the date. Finally, the maximum and minimum normalization method is used to integrate the data into (0, 1) range data, which eliminates the influence between samples of different orders of magnitude. The maximum and minimum normalization formula is as follows

$$X_{\mathrm{norm}}=\frac{X-X_{min}}{X_{max}-X_{min}}.$$

(3)

The distribution graph and Q-Q graph of the processed data are shown in Figure 7 and Figure 8 respectively. As can be seen from the figure, the data has basically satisfied the normal distribution.

This data set contains feature data related to the number of new crowns, irrelevant feature data and related but redundant feature data. In the face of complex faults, it is no longer possible to accurately obtain the number of new crowns by relying only on expert experience and simple correlation analysis to perform feature selection work. Important features, so this article uses random forest (RF) out-of-bag estimation to rank the importance of new crown-related features. The random forest is used to select the features of the data set, and the features that have little influence on the prediction results are eliminated.

RF is a combined classifier based on decision trees, which can be used for feature selection [11]. RF uses the Bagging method to randomly and repeatably extract samples from the original sample set for classifier training. About 1/3 of the sample data will not be selected [12]. This data is called Out of Bag (OOB). When calculating the importance of a certain feature, use the OOB data as the base learner after the test set to test the training, and the test error rate is recorded as the out-of-bag error (errOOB). Add noise to the important features to be calculated in the OOB sample, and recalculate errOOB again. The average test error of all base learners is calculated by using the average accuracy decrease rate (MDA) as an indicator for feature importance calculation, namely

$$MDA=\frac{1}{n}\sum _{t=1}^n\left(errOO{B}_t-{errOOB}_t^{\prime }\right),$$

(4)

where n is the number of base learners, errOOB is the out-of-bag error after adding noise.

The more the MDA index decreases, the more the corresponding feature has a greater impact on the prediction result, and the higher its importance. This feature importance calculation method is called random forest out-of-bag estimation. According to this method, the importance of fault-related features is ranked and feature selection is performed.

2.2.3. Tuning Algorithm

For the LightGBM model, there are many internal hyperparameters that affect the prediction results. However, if the value of the hyperparameter used is the default value, this hyperparameter combination may not be the optimal hyperparameter combination for the new coronavirus number prediction data set [13]. Therefore, this paper introduces three tuning algorithms, namely grid search, random search, and Bayesian optimization, to optimize some important hyperparameters of LightGBM [14]. Before adjusting the parameters of LightGBM, the optimization range of hyperparameters is generally set first. These three algorithms are briefly described below.

Grid search divides the search range into grid shapes, and adjusts the parameters according to the set step to train the model until all possible combination parameters are verified, and finally the parameter combination that gives the best result is output [15]. Because the different prediction results of the data in each group of hyperparameter combinations are also different, when the hyperparameter combination is relatively large and the search range is relatively large, the optimization speed of the grid search is very slow.

Random search is similar to grid search, but it does not verify all possible parameter combinations like grid search, but randomly combines the random value of each parameter, so the speed of random search is faster than that of Grid search [16]. However, random search may also miss the parameter combination that maximizes the prediction result.

Bayesian optimization algorithm(BOA) can quickly find the optimal parameters for the problem to be solved based on historical experience [17]. The main problem scenarios for Bayesian optimization are

$$X^{*}=argmaxf\left(x\right)(x\in S),$$

(5)

where x is the parameter to be optimized, S is the candidate set of x variable, that is, the set of possible values of parameter x. The target selects an x from the set S such that the value of $f\left(x\right)$ is the largest or smallest. Here, the specific formula of $f\left(x\right)$ may not be known, that is, the black box function. But you can choose an x, and get the value of $f\left(x\right)$ through experiment or observation [18].

BOA has two core processes, a priori function (PF) and acquisition function (AC). The acquisition function is also called the efficiency function. Under the framework of Bayesian decision theory, many collection functions can be interpreted as evaluating the expected loss associated with f at point x, and then usually selecting the point with the lowest expected loss [19]. PF mainly uses Gaussian process regression, AC mainly uses these methods including EI (expected improvement), PI (probability of improvement) and UCB (upper confidence bound), and this article uses the EI function. The EI function can find out the global optimum without falling into the local optimum. The collection function is as follows

$$u\left(x\right)=max\left(0,f^{\prime }-f\left(x\right)\right),$$

(6)

where f is the collection function, and $f\left(x\right)$ is the optimized performance indicator.

The final collection function for variable x is

$$\begin{array}{c}{a}_{\mathrm{EI}}\left(x\right)=\mathrm{E}[\mathbf{u}\left(x\right)\mid x,D]={\int }_{-\infty }^{f^{\prime }}\left(f^{\prime }-f\right)\mathrm{N}(f;\mathbf{u}\left(x\right),K(x,x))df\\ \left.=\left(f^{\prime }-u\left(x\right)\right)\Phi \left(f^l;u\left(x\right),K(x,x)\right)+K(x,x)\right)N\left(f^{\prime };u\left(x\right),K(x,x)\right).\end{array}$$

(7)

The calculation shows that the point corresponding to the maximum value of ${\mathbf{a}}_{\mathrm{EI}}$ is the best point. There are two components in Formula (7). To maximize the value of it, you need to optimize the left and right parts at the same time, that is, the left side needs to reduce the $\mu \left(x\right)$ as much as possible, and the right side needs to increase the variance (or covariance) K(x, x) as small as possible. It is a typical theory on issues such as exploration and exploitation.

Upper confidence bound (UCB) can be simply understood as the upper confidence boundary. It is usually described by maximizing f instead of minimizing f. But in the case of minimization, the collection function will take the following form

$$a_{UCB}\left(x\right)=u\left(x\right)-\beta \sigma \left(x\right),$$

(8)

where $\beta >0$ is a strategy parameter, and $\sigma \left(x\right)=\sqrt{K(x,x)}$ is the boundary standard deviation of $f\left(x\right)$. Similarly, UCB also includes exploitation $\left(u\right(x\left)\right)$ and exploration $\left(\right(x)$ modes. It can converge to the global optimal value under certain conditions.

Table 3. The LightGBM hyperparameters selected in this article and their functions.

Parameter	Style	Search Scope	Effect
learn_rate	float	(0.001, 0.3)	improve accuracy
max_depth	int	(3, 10)	prevent overfitting
num_leaves	int	(3, 1024)	improve accuracy
min_data_in_leaf	int	(0, 80)	prevent overfitting
feature_fraction	float	(0.2, 0.9)	accelerate
bagging_fraction	float	(0.2, 0.9)	accelerate
lambda_l1	float	(0, 10)	prevent overfitting

shows the hyperparameter combinations selected in this article and the corresponding descriptions.

2.2.4. LightGBM

LightGBM is an open source decision tree-based gradient boosting framework proposed by Microsoft. As an improved version of Gradient Boosting, it has the characteristics of high accuracy, high training efficiency, support for parallelism and GPU, small memory required, and ability to handle large-scale data [20].

According to the different generation methods of the base learner, integrated learning can be divided into parallel learning and serial learning. As the most typical representative of serial learning, Boosting algorithm can be divided into Adaboost and Gradient Boosting. The main difference between them is that the former improves the model by increasing the weight of misclassified data points, while the latter improves the model by calculating negative gradients. The core idea of Gradient Boosting is to use the negative gradient of the loss function to approximate the value of the current model $f\left(x\right)=f_{j-1}\left(x\right)$ to replace the residual. Suppose the training sample is i ($i=1,2,\dots ,n$), the number of iterations is j ($j=1,2,\dots ,m$), and the loss function is $L(yi,f(xi\left)\right)$, then the negative gradient $r_{ij}$ can be expressed as

$$r_{ij}=-{\left[\frac{\partial L\left(y_i,f\left(x_i\right)\right)}{\partial f\left(x_i\right)}\right]}_{f\left(x\right)=f_{j-1}\left(x\right)}.$$

(9)

Use the base learner $h_j\left(x\right)$ to fit the negative gradient r of the loss function, and find the best fit value $r_j$ that minimizes the loss function

$$r_j=argminL\left(y_i,f_{j-1}\left(x_i\right)+r{h}_j\left(x_i\right)\right).$$

(10)

Model update:

$$\mid f_j\left(x\right)=f_{j-1}\left(x\right)+r_j{h}_j\left(x\right).$$

(11)

Gradient Boosting generates a base learner in each round of iteration. Through multiple rounds of iteration, the final strong learner $F\left(x\right)$ is the base learner generated in each round and obtained by linear addition:

$$F\left(x\right)=f_m\left(x\right)$$

(12)

As an improved lightweight Gradient Boosting algorithm, the core ideas of LightGBM are: histogram algorithm, leaf growth strategy with depth limitation, direct support for category features, histogram feature optimization, multithreading optimization, and cache hit rate optimization. The first two features effectively control the complexity of the model and realize the lightweight of the algorithm, so this article is particularly concerned.

The histogram algorithm discretizes continuous floating-point features into L integers to construct a histogram with a width of L. When traversing the data, use the discretized value as an index to accumulate statistics in the histogram. After traversing the data once, the histogram accumulates the necessary statistics, and then find the optimal split point from the discrete values of the histogram .

The traditional leaf growth strategy can split the leaves of the same layer at the same time. In fact, the splitting gain of many leaves is low and there is no need to split, which brings a lot of unnecessary expenses. For this, LightGBM uses a more efficient leaf growth strategy: each time it searches for the leaf with the largest split gain from all the current leaves to split, and sets a maximum depth limit. While ensuring high efficiency, it also prevents the model from overfitting.

3. Results and Discussion

3.1. Performance Predictor

All models are cross-validated and the coefficient of determination (R2), mean absolute error (MAE), relative absolute error (RAE), relative square root error (RRSE), root mean square error (RMSE) are calculated, as shown below

$$R2(y,\widehat{y})=\sqrt{1-\frac{{\sum }_{i=1}^n{\left(y_i-{\widehat{y}}_i\right)}^2}{{\sum }_{i=1}^n{\left(y_i-\overline{y}\right)}^2}},$$

(13)

$$RMSE(y,\widehat{y})=\sqrt{\frac{1}{n}\sum _{i=1}^n{y}_i-{\widehat{y}}_i^2},$$

(14)

$$MAE(y,\widehat{y})=\frac{1}{n}\sum _{i=1}^n\left|y_i-{\widehat{y}}_i\right|,$$

(15)

$$RAE(y,\widehat{y})=\sqrt{\frac{{\sum }_{i=1}^n\left|y_i-{\widehat{y}}_i\right|}{{\sum }_{i=1}^n\left|y_i-\overline{y}\right|}},$$

(16)

$$RRSE(y,\widehat{y})=\sqrt{\frac{{\sum }_{i=1}^n{\left(y_i-{\widehat{y}}_i\right)}^2}{{\sum }_{i=1}^n{\left(y_i-\overline{y}\right)}^2}},$$

(17)

where y represents the true value, $\widehat{y}$ represents the predicted value, $\overline{y}$ represents the average value of the true value and n is the number of test sets.

3.2. Experiment Results

Figure 9 shows the result of feature selection using random deep forest, and the features are output in descending order of importance. It can be seen from the figure that Death Toll has the greatest impact on Diagnosis Numbers, while the attribute of Month has the least impact. Finally, we selected the 7 most influential attributes for the prediction of Diagnosis Numbers.

According to the optimal parameter set of the model, the Diagnosis Numbers prediction model of COVID-19 is constructed. In this paper, LightGBM, GridSearch-LightGBM, RandomSearch-LightGBM, and BOA-LightGBM models are used for Diagnosis Numbers prediction.

Table 4. Specific parameter values found by three tuning algorithms.

Parameter	GridSearch	RandomSearch	BOA
learn_rate	0.632	0.828	0.355
max_depth	7	8	5
num_leaves	225	237	249
min_data_in_leaf	33	27	30
feature_fraction	0.7	0.7	0.8
bagging_fraction	0.7	0.7	0.8
lambda_l1	2.34	3.45	1.80

Note: BOA = Bayesian optimization algorithm.

shows the specific values of the optimal parameter combinations found by the three tuning algorithms.

Table 5. Model evaluation index.

Models	R2	RMSE	MAE	RAE	RRSE
LightGBM	0.820	354.945	138.939	0.535	0.424
GridSearch-LightGBM	0.865	311.918	145.266	0.548	0.368
RandomSearch-LightGBM	0.861	316.217	137.621	0.533	0.373
BOA-LightGBM	0.879	295.686	124.911	0.508	0.348

Note: GBM, gradient boosting machine; BOA, Bayesian optimization algorithm; R2, coefficient of determination; RMSE, root mean square error; MAE, mean absolute error; RAE, relative absolute error; RRSE, relative square root error.

shows the evaluation indicators of the prediction results of the four models. The prediction results of the model are evaluated by R2, RMSE, MAE, RAE, RRSE evaluation indicators. It can be seen from the values of the five evaluation indicators that the results of BOA-LightGBM are better than the former. RandomSearch-LightGBM and GridSearch-LightGBM have their own advantages and disadvantages. It can also be seen that the default hyperparameters of LightGBM are not suitable for the prediction of Diagnosis Numbers of COVID-19 in this article. From the approximate prediction effect, BOA-LightGBM can better analyze the relationship between historical data and can effectively predict the value of Diagnosis Numbers of COVID-19, which proves the superiority of the model.

Figure 10 is a line chart of the four algorithms to predict Diagnosis Numbers, and only part of the data is taken on the abscissa. The prediction effect of the model can be seen more intuitively from the line graph. It can be seen from the figure that in most cases, the BOA-LightGBM model can better fit the fluctuation trend of Diagnosis Numbers at some points, and the predicted value is very close to the actual value. In the figure, the points predicted by GridSearch-LightGBM are basically covered, so they are not shown in the figure, which just shows that the prediction results are not very prominent. Sometimes the prediction value of LightGBM is better than other models, but most of them are inferior to other models. So comprehensively, the BOA-LightGBM model is more in line with the changing trend of real values.

4. Conclusions

This study uses the Internet big data tool-Baidu Index to predict the development trend of the new coronavirus pneumonia epidemic to obtain data. By selecting appropriate keywords, data on COVID-19 cases in China from 1 January 2020 to 1 April 2020 are collected. After preprocessing the data set, the random forest feature selection method is used to obtain the optimal sub-data set. After comparing and analyzing the optimization results of the seven hyperparameters of the LightGBM model with the three optimization algorithms of grid search, random search, and Bayesian optimization. It is concluded that applying the data set obtained from the Baidu Index to the Bayesian-optimized LightGBM model can better predict the increase in the number of new coronary pneumonias, and it is a good aid to predict the new number of new coronary pneumonia in the future medical structure effect.