HUBEI AGRICULTURAL SCIENCES ›› 2024, Vol. 63 ›› Issue (8): 116-120.doi: 10.14088/j.cnki.issn0439-8114.2024.08.020

• Production and Growth Model • Previous Articles     Next Articles

A small sample based method for predicting nitrogen application rates in wheat

DU Yun, ZHANG Jing-jing, HAN Bo, LU Zi-ao   

  1. College of Computer and Information Engineering/Engineering Research Center of Intelligent Agriculture, Ministry of Education/Xinjiang Agricultural Informatization Engineering Technology Research Center, Xinjiang Agricultural University , Urumqi 830052, China
  • Received:2024-03-25 Online:2024-08-25 Published:2024-09-05

Abstract: A XGBoost algorithm prediction model based on SBS (SMOTE+Bootstrap) data augmentation method was proposed to address the problem of limited data on fertilization experiments during the growth cycle of wheat (Triticum aestivum L.) and difficulty in effectively predicting fertilization using traditional prediction methods. Based on the original 135 nitrogen application data, the training set (80%) and the test set (20%) were divided. The SMOTE method was used to balance the training and test sets to obtain more feature information. Then, the Bootstrap method was used to expand the balanced data. Finally, the XGBoost prediction model was used for training and compared with other machine learning models. The results showed that using the SMOTE method to balance data significantly improved the prediction accuracy of the SBS-XGBoost model. MSE decreased from the original data of 66.802 to 13.027, MAE decreased from the original data of 6.711 to 2.393, and R2 increased from the original data of 0.390 to 0.912. SBS-XGBoost not only performed well in predicting nitrogen application rates in this study, but also provided reference and guidance for scientific prediction of other small sample data.

Key words: wheat (Triticum aestivum L.), small sample, nitrogen application rate, prediction

CLC Number: