An Improved CatBoost-Based Classification Model for Ecological Suitability of Blueberries

Abstract

Selecting the best planting area for blueberries is an essential issue in agriculture. To better improve the effectiveness of blueberry cultivation, a machine learning-based classification model for blueberry ecological suitability was proposed for the first time and its validation was conducted by using multi-source environmental features data in this paper. The sparrow search algorithm (SSA) was adopted to optimize the CatBoost model and classify the ecological suitability of blueberries based on the selection of data features. Firstly, the Borderline-SMOTE algorithm was used to balance the number of positive and negative samples. The Variance Inflation Factor and information gain methods were applied to filter out the factors affecting the growth of blueberries. Subsequently, the processed data were fed into the CatBoost for training, and the parameters of the CatBoost were optimized to obtain the optimal model using SSA. Finally, the SSA-CatBoost model was adopted to classify the ecological suitability of blueberries and output the suitability types. Taking a study on a blueberry plantation in Majiang County, Guizhou Province, China as an example, the findings demonstrate that the AUC value of the SSA-CatBoost-based blueberry ecological suitability model is 0.921, which is 2.68% higher than that of the CatBoost (AUC = 0.897) and is significantly higher than Logistic Regression (AUC = 0.855), Support Vector Machine (AUC = 0.864), and Random Forest (AUC = 0.875). Furthermore, the ecological suitability of blueberries in Majiang County is mapped according to the classification results of different models. When comparing the actual blueberry cultivation situation in Majiang County, the classification results of the SSA-CatBoost model proposed in this paper matches best with the real blueberry cultivation situation in Majiang County, which is of a high reference value for the selection of blueberry cultivation sites.

Assessing Forest Quality through Forest Growth Potential, an Index Based on Improved CatBoost Machine Learning

“…CatBoost is a machine-learning library opened by the Russian search giant Yandex in 2017; it has become one of the three mainstream artifact libraries of gradient-boosting decision trees (GBDTs) together with lightGBM and xgBoost [21][22][23]. It has a GBDT framework based on the oblivious trees algorithm, which has few parameters, supports categorical variables, and has high accuracy.…”

Identifying the Biological Characteristics of Anthracnose Pathogens of Blueberries (Vaccinium corymbosum L.) in China

“…Their preference for acidic soils often leads to their presence in coniferous forests [1,2]. In China, blueberries are found mainly in the forested areas of the Daxinganling and Xiaoxinganling mountains, as well as in the middle and lower reaches of the Yangtze River and in the south [3]. Within forest ecosystems, blueberries serve as a vital food source for various wildlife species, such as birds, bears, and small mammals.…”