Evaluation of machine learning approaches for prediction of pigeon pea yield based on weather parameters in India

Pigeon pea is the second most important grain legume in India, primarily grown under rainfed conditions. Any changes in agro-climatic conditions will have a profound influence on the productivity of pigeon pea (Cajanus cajan) yield and, as a result, the total pulse production of the country. In this context, weather-based crop yield prediction will enable farmers, decision-makers, and administrators in dealing with hardships. The current study examines the application of the stepwise linear regression method, supervised machine learning algorithms (support vector machines (SVM) and random forest (RF)), shrinkage regression approaches (least absolute shrinkage and selection operator (LASSO) or elastic net (ENET)), and artificial neural network (ANN) model for pigeon pea yield prediction using long-term weather data. Among the approaches, ANN resulted in a higher coefficient of determination (R² = 0.88-0.99), model efficiency (0.88-1.00) with subsequent lower normalised root mean square error (nRMSE) during calibration (1.13-12.55%), and validation (0.33-21.20%) over others. The temperature alone or its interaction with other weather parameters was identified as the most influencing variables in the study area. The Pearson correlation coefficients were also determined for the observed and predicted yield. Those values also showed ANN as the best model with correlation values ranging from 0.939 to 0.999 followed by RF (0.955-0.982) and LASSO (0.880-0.982). However, all the approaches adopted in the study were outperformed the statistical method, i.e. stepwise linear regression with lower error values and higher model efficiency. Thus, these approaches can be effectively used for precise yield prediction of pigeon pea over different districts of Karnataka in India.