Crop Yield Prediction of Indian States Based on Machine Learning and Feature Engineering
摘要
Crop yield prediction for Indian states is a significant topic in agriculture, utilizing machine learning and statistical models to forecast production. It helps farmers, policymakers, and researchers optimize resources and improve food security. This study aims to analyse and predict crop yield across various states of India using machine learning approaches. This study comprises agricultural data for a variety of crops (55 different crops) cultivated across different states (30 states) in India, spanning the years 1997 to 2020. It includes key features essential for crop yield prediction, such as crop type, crop year, cropping season, state-wise cultivation areas, production volumes, annual rainfall, fertilizer and pesticide usage, and calculated crop yields. This study aims to develop feature engineering based machine learning models to predict crop yield of top ten crops of Indian States. This study has tested and compared the performance of ML models (DT, RF, SVR, XGB and DNN) across new set of features (derived features) on top ten different crops. Among all the models, XGBoost and DNN outperform others in capturing nonlinear relationships, especially for high-yielding crops like Sugarcane and Wheat.