Plant-based sensing on water stress can provide sensitive and direct reference for precision irrigation system in greenhouse. However, plant information acquisition, interpretation, and systematical application remain insufficient. This study developed a discrimination method for plant root zone water status in greenhouse by integrating phenotyping and machine learning techniques. Pakchoi plants were used and treated by three root zone moisture levels, 40%, 60%, and 80% relative water content. Three classification models, Random Forest (RF), Neural Network (NN), and Support Vector Machine (SVM) were developed and validated in different scenarios with overall accuracy over 90% for all. SVM model had the highest value, but it required the longest training time. All models had accuracy over 85% in all scenarios, and more stable performance was observed in RF model. Simplified SVM model developed by the top five most contributing traits had the largest accuracy reduction as 29.5%, while simplified RF and NN model still maintained approximately 80%. For real case application, factors such as operation cost, precision requirement, and system reaction time should be synthetically considered in model selection. Our work shows it is promising to discriminate plant root zone water status by implementing phenotyping and machine learning techniques for precision irrigation management.
Over the past decade, the environmental information based methods for precision management have been widely applied in greenhouse practice. The ambient climate of greenhouse such as temperature, humidity, and root zone water content is monitored closely as the reference for crop management1. Water and fertilizer are the most important inputs for greenhouse crop because the plant growth is significantly affected by plant water and nutrition status. Root zone water status has a significant influence on plant evapotranspiration rate2. Root zone water status indicates the moisture situation of plant root growth environment and highly correlates to plant growing status3. Monitoring of root zone water status could help to establish the precision management strategy on irrigation in crop production. However, conventional probe sensors for moisture measurement are limited on detecting the actual substrate water status effectively in trays. Several commercialized soil moisture sensors such as tensiometer, neutron probes and time domain reflectometry probes have been widely used in pot planting but not applicable for seedlings mainly because the high costs, unsuitable size, and unreliable measurements. The volume of cell in plug tray is minimal resulting that common probe sensors can hardly insert into substrates. In addition, due to the uneven water distribution among different plug trays, the growers require continuous and real-time water content monitoring for large -scale rather than for single point, to prevent the seedlings from water stress and growth restriction. The relationship between plug tray weight and root zone water content was studied to provide a weighting based irrigation control strategy4. However, irrigation based on tray weight is labor intensive and inaccuracy in practice.
With the development of sensing technology, real-time monitoring of physiological and ecological information of the plant itself provides an approach to indicate root zone water status5, 6. Plant physiological traits such as tissue water status, stomatal conductance, and sap flow are known sensitively responding to water stress. The status of plant water content could be a useful indicator for irrigation7. Phenotypic traits such as leaf area, leaf angle, and thermal temperature show strong correlations with the root zone water status8. However, most of the plant -based irrigation methods are still at research/developing stage and little used yet for practice (except for thermal sensing in some situations). Systematical research on phenotypic traits selection for irrigation has seldom been reported.
Rapid and non-destructive plant sensing method is needed for plant-based information acquisition. The application of machine vision technology could be the solution. Plant phenotypic traits can be quickly and efficiently obtained by imaging-based automatic inspection and analysis5. Therefore, machine vision has the potential to realize the rapid, reliable, real-time monitoring of plant status with low cost, which meets the requirements for precision irrigation scheduling9. Machine vision technology has been applied in plant monitoring such as the detection of plant abiotic stress through color variation, plant movement monitoring, textural features detection, canopy temperature measurement and detection of plant biotic disease8, 10,11,12,13. With the development of phenotyping technology, various imaging sensors have been used to collect phenotypic parameters which are related to plant stress in automated or semi-automated high-throughput plant phenotyping platforms14, 15. However, data generated by these platforms has enormous volume, variety, velocity, and veracity, which must be efficiently analyzed. The classical statistical method is difficult to analyze the high-dimensional and dynamical image data, resulting limited phenotyping interpretation16.
Machine learning, a type of artificial intelligence has good performance in efficiently dealing with large amount of data in various domains. It is promising in phenotyping applications on features extraction and patterns identification from large dataset through holistic approaches. Machine learning algorithm has few input factors with high predict accuracy and strong adaptability. Several machine learning algorithms, such as support vector machine (SVM), artificial neural network (ANN) and random forest (RF), have been used for classification model development17, 18.
In practice, machine learning method has been applied to remote sensing system and irrigation system19, 20. Many studies have reported plant stress phenotyping using machine learning in identification, classification, quantification, and prediction of plant water status, but few focused on phenotyping root zone water stress10, 21, 22. It is crucial to detect plant water stress at the early stage before any damage observed with visible wilting, which affect subsequent plant growth and quality. Correlating root zone water status with plant phenotype to supervise irrigation system could be a practical way in precision irrigation system with easy operation and quick response.
Pakchoi (Brassica campestris ssp. L. chinensis) is one of typical vegetables in greenhouse production, which is sensitive to water status. During the past few years, commercial production of leafy vegetable has been initiated and developed both in the greenhouse (solar plant factory) and artificial light plant factory23. Growing baby pakchoi with plug trays using substrate in greenhouse is a promising production mode because it is suitable for mechanized harvesting and precision management, which improves the production efficiency24. The accurate irrigation scheduling is critical for safe and high quality of greenhouse vegetable, as well as improving the resource utilization and production profits. This study aims to discriminate plant root zone water status nondestructively, taking pakchoi as the plant material, which is a part of on-going research on developing machine vision-based precision irrigation system for factory production of pakchoi. The objectives of this study are: (1) investigating the relationship between root zone water status and plant growth; (2) developing a root zone water status discrimination model by applying phenotyping technology and machine learning algorithms; (3) evaluating the performance of developed models in various application scenarios.
Results and Discussion
The effect of root zone water status on plant growth
The effect of root zone water status on pakchoi plant growth was investigated under three relative water content treatments, 40% (low), 60% (medium), and 80% (high). The fresh and dry weight of shoot and root for both cultivars were significantly different between treatments as shown in Fig. 1 with ANOVA test (p < 0.01). Treatments had different effects between cultivars. For cultivar “Huawang”, treatment of medium water content had the highest values of fresh and dry weight of shoot and root (fresh weight as 2727.1 mg and 184.4 mg for shoot and root respectively, and dry weight as 185.1 mg and 15.1 mg per plant for shoot and root respectively), while cultivar “Kangre 605” had the lowest values under such treatment (fresh weight as 500.3 mg and 52.0 mg for shoot and root respectively, and dry weight as 45.3 mg and 5.1 mg per plant for shoot and root respectively). The differences of pakchoi growth were significant between two cultivars under the treatment of medium water content (p < 0.01 for all comparisons), but there were no significant differences observed between cultivars under the other two treatments. This is mainly because “Kangre 605” is a heat-resistant cultivar, which has the ability of growing well under water stress conditions. “Huawang” prefers moderate water condition, and both the excess and deficiency of water supply could affect the accumulation of biomass.
Relationship between root-zone water status and phenotypic traits
Phenotypic traits extraction
Two types of images were acquired, visible and near-infrared (NIR) image (Supplementary Fig. S1). There were in total 37 phenotypic traits extracted from images for each plant (Table 1). These phenotypic traits were classified into three categories for easy description, morphological trait, color trait, and NIR trait, including 19, 6, and 12 traits respectively. Morphological traits mainly indicated the shape feature of target plant region, such as circumference and eccentricity. Color traits were generated from color images, including the pixel value information of R (red), G (green), and B (blue) components. NIR traits were the pixel intensity information of different ranges in NIR images.
Water related phenotypic traits selection
In order to increase the efficiency of model development, traits selection is necessary to identify redundant traits which are irrelevant to root zone water status. The significance of 37 extracted phenotypic traits on classifying root zone water status were investigated by using ANOVA (Fig. 2). For better illustration, the p value was transformed into -log10 (p). There were five traits with p > 0.05 (i.e. −log10 (p) < 1.301) including Normsmallpax, Circumference, Roundness, Bdrycount and Bdrytoarearatio (numbered as 14, 23, 26, 28, and 29 in Table 1 respectively). Therefore, the five traits were excluded from the datasets in the following modeling process.
Root zone water status discrimination model development and validation
Modeling potentiality assessment
The Partial Least Squares - Discriminant Analysis (PLS-DA) was conducted to present a visual space distribution of three treatment groups for assessing the modeling potentiality using the 32 selected phenotypic traits. Figure 3 shows the PLS-DA model performance and cumulative variance explained by the top five components. The goodness of fit was quantified by R2 while the predictive ability was indicated by Q2. The accuracy, R2, and Q2 of PLS-DA model by using all the top five components reached 73%, 0.35, and 0.34, respectively. The value of R2 and Q2 should be greater than 0.5 to be taken as an indicator of model acceptability. The cumulative explained variance ranged from 17.6% to 72.2% by the number of components from one to five, with the first three components explaining 66.9% variance. The performance of PLS-DA model improved (R2 = 0.67, Q2 = 0.66) by excluding the medium root zone water group as the difference between medium water group and the other two groups was not significant. Overall, PLS-DA shows the potential of root zone water status discrimination by phenotypic traits.
Discrimination model development
The total dataset used for model development was 2120 samples with 32 traits of each (2120*32), and the dataset was separated into 80% training dataset and 20% testing dataset. The training data set was used for model development. Parameter values for the three modeling algorithms, which are Random Forest (RF), Neural Network (NN), and Support Vector Machine (SVM), were selected by using 10-fold cross validation in three repetitions tuning (Table 2). Among the training dataset, 10% of the data were used as validation in each cross validation. Table 2 also lists the package, function, and program training time of modeling algorithms. All the R codes ran in Rstudio 1.0.44 on the platform of windows 10 × 64 system with an Intel Core i7-5500U CPU and 4 GB RAM.
Considering the dynamics of the data input and training in practice, the model training time is an important factor to be considered in the optimization process. The training time increases exponentially with the number of parameters and varies substantially between algorithms. Among these three modeling algorithms, NN took the shortest training time as 544.37 seconds, which was approximately 76% of RF and 44% of SVM. The training time consumption of SVM should be considered in real case of decision support system as this method is more sensitive to the data size than others, especially when time is limited and data set is large. The trait contribution analysis in each model could be a way to simplify the model inputs and reduce the training time, which was discussed in section 3.3.
The mean prediction accuracy and Cohen’s Kappa coefficient (k) of all the three models were higher than 90% and 0.85 respectively, as shown in Fig. 4A,B. SVM model had the best performance with higher accuracy and k value (92.5% and 0.89 respectively) than the other two models. The result also showed that SVM was slightly more stable with less variation than others.
The ROC (receiver operating characteristic) curve was used to evaluate the performance of modeling classifier. Figure 4C shows the ROC curves of the three models. The x-axis represents false positive rate (FPR), and the y-axis represents true positive rate (TPR). The area under the curve (AUC) indicates classification performance, which is the ability of target model to correctly classify the plants in different root zone water status. SVM model had the highest AUC value as 0.6827 among the three models, while the differences were not substantial compared to 0.6766 and 0.6788 AUC of RF model and NN model respectively.
Discrimination model validation
Confusion matrix was implemented to validate the three developed models with 2120*32 dataset. The prediction accuracy of each model from cross validation is the sum of the prediction accuracy on the diagonal in each sub-table of Table 3. All the developed models had good prediction accuracy higher than 90%. SVM model had the highest prediction accuracy as 92.5%, but no substantial difference observed on accuracy among the three models (91.3% and 91.4% for RF model and NN model respectively).
Compared to the other two treatments, the medium root zone water group had the highest percentage of false positive (4.0%, 3.4%, 3.8% for RF, NN, and SVM model) and false negative (3.2%, 2.9% and 3.2% RF, NN, and SVM model) among the three models. It could be explained as that medium level is adjacent to both high and low level, which is less distinguishable comparing to the other two levels.
Model evaluation under different conditions
To investigate the robustness and applicability of the three models in different conditions, the test dataset was reorganized into different scenarios, different time of a day, growth stage (stage 1 to stage 5), weather condition (sunny and cloudy), and cultivar (“Huawang” and “Kangre 605”). The dataset used for each growth was 120*32, the dataset used for each time in day was 240 to 270 records with 32 traits, and the dataset used for each cultivar was 1060*32. All the datasets were separated into 80% training dataset and 20% testing dataset. Figure 5 shows classification accuracy of three models in all scenarios were over 85%. Among the seven time points of a day, the accuracy of the three models were relatively lower at 06:00 h than at other time points (Fig. 5a), which might be caused by the rapid decline of the greenhouse humidity at sunrise resulting in the change of leaf surface moisture status. For NN model, its accuracy dropped at 10:00 h and 14:00 h with the highest accuracy at noon, while RF and SVM model had high accuracy at 10:00 h as 100.0%. For different growth stage, the accuracy of the three models had the trend of increasing (Fig. 5b). The accuracy of RF and SVM model reached 100.0% in the fourth and fifth growth stage which were substantially higher than NN model. In the early stage, leaf area was small resulting in poor discrimination performance. With leaf expanding, the difference of traits between groups increased. In addition, as the treatments of root zone water moisture level was applied during the entire growing period, it would have accumulative effect on plant phenotype, which increased the model classification accuracy in later stages.
The accuracy of the three model were slightly higher in cloudy weather condition than in sunny weather condition (Fig. 5c), which might be because of less environmental fluctuation in cloudy day. SVM model had the highest accuracy in both weather conditions (91.1% and 94.1% respectively). Among the three models, NN model had the largest difference of accuracy between cultivar (5.8%, Fig. 5d) compared to RF and SVM model (1.1% and 3.2% respectively). Overall, RF model performed more stable than the other two considering all the scenarios.
Trait contribution analysis in developed models
In order to check the possibility of reducing phenotyping cost and model training time without significant accuracy reduction, sensitivity analysis was conducted to assess the contribution of each trait in developed models. The simplified model with less explanatory variables were developed according to the traits importance evaluation of the model implementation in the root zone water status detection. Table 4 lists the top five of most contributing traits in each model. The explanation of each trait could be referred to Table 1. Different model had different preference on the trait type. In the top five traits, there was no morphological trait in RF model, no NIR trait in NN model, and no color trait in SVM model.
Overall, color trait and NIR trait appeared more frequently than morphological trait, which shows plant color and NIR characteristics would be more related to root zone water status. This result agrees with several studies in spectroscopy and remote sensing research on the relationship between spectral reflectivity and water related index13. The most frequently appearing traits in top five traits of developed models were Mean.Color.Green and Mean.Color.Red. There was only one morphological trait in NN model and in SVM model, named as Vrectsizeymm and Compactness, respectively. This is likely because plant morphological appearance is basically determined by genotype and substantially influenced by environment such as airflow and illumination.
Model classification ability dropped when only the top five traits were used to develop model, named as simplified model. The accuracy of simplified RF model and NN model could still reach 79.7% and 77.4% respectively, while the accuracy of simplified SVM model decreased substantially from 92.5% to 63.0% with 29.5% reduction. For SVM model, the cumulative contribution of the top five traits was 22.8%, lower than that of RF model and NN model (32.6% and 29.9%, respectively), which could be the main reason for the great decrease on the simplified model accuracy. In addition, simplified SVM model did not include color trait, while simplified NN model did not include NIR trait (Table 4), which showed color related traits would better reflect plant root zone water status than the other two trait types in RF and NN models while the SVM model relied more on NIR traits than the others.
This study developed a root zone water status discrimination method during plant growth by integrating phenotyping and machine learning techniques. Pakchoi plants were used. Three types of phenotyping traits such as morphological trait, color trait, and near-infrared trait were acquired as the inputs for classification models. Three machine learning models were developed, Random Forest (RF), Neural Network (NN), and Support Vector Machine (SVM), with accuracy above 90% for all. The SVM model had the highest accuracy as 92.5% in model development but it took the longest training time as 1245.94 seconds in this study. In addition, the SVM model had the largest accuracy reduction (22.8%) in simplified model developed by the top five most contributing trait resulting from trait contribution evaluation.
The three developed models were evaluated by test dataset (20% of entire data pool) consisting of data from different scenarios, which were different time a day, growth stage, weather condition, and pakchoi cultivar. All the three models reached the overall accuracy higher than 90%. SVM model had the highest value as 92.5%, but no substantial differences were observed among the three. By evaluated in different scenarios, the three models had accuracy over 85% in all scenarios with some fluctuations. The accuracy at the time point of 06:00 h was the lowest for all the three models, and the trend of increasing was observed during the growth of plant. Better performances were observed in cloudy weather condition than in sunny weather condition of the three models. NN model had the largest difference on accuracy between cultivars. Overall, SVM model had the highest classification accuracy, but more stable performance was observed in RF model considering all the scenarios.
This study demonstrates the potential of machine learning approach on discriminating root zone water status based on complex plant phenotyping traits. The developed discrimination method could promote the plant-based irrigation decision making and implementation in practice. Developed models could be further used in precision irrigation system with appropriate modifications if environment or crop change. Further study is suggested to investigate the repaid response of plant phenotypic traits to the change of root zone water status and the performance of classification models on such response. Therefore accumulative effect of water stress on plant growth during the growing period should be minimized with the real-time discrimination to control the potential yield loss. Population phenotyping and water status discrimination are also worth further study due to plant canopy mutual occlusion and growth competition. In addition, phenotyping traits were extracted from images acquired in a constant light environment. Changes in intensity, quality, and beam angle of illumination would have effects on imaging process and hence affect the modeling process. Therefore, research on effective acquisition of phenotyping traits in natural light environment would be a direction to reduce cost and enhance applicability. Our presented work is capable of serving as a basis and supporting for intelligent greenhouse management, especially for irrigation management.
Materials and Methods
Plant materials and growth conditions
The experiment was conducted in a commercial greenhouse from May 1th to 29th in year of 2015, located at northeast of Shanghai, China (31°10′ N, 121°36′ E). Two cultivars of pakchoi were used, “Huawang” and “Kangre 605”, under pot substrate cultivation. Commercial substrate (vegetable seedling substrate, Zhongnuo Agriculture Technology Co., Ltd, Huaian, China) composed of peat moss, perlite, and organic matter was used. Plug tray was used at seedling stage. On the tenth day after sowing, seedlings were transplanted to pots (volume of 1.13 L) with one seedling per pot.
The greenhouse air temperature, humidity, photosynthetic photon flux, substrate temperature and electrical conductivity were monitored every 5 minutes by using an automatic data logging system (PM-11 Phytomonitor, Bio Instruments S.R.L., Chisinau, Moldova) with RTH-11 Meter and SMTE sensor. During the whole experimental period, the average air temperature was 24.3 °C with the highest as 36 °C and lowest as 13.7 °C. The relative humidity ranged from 16.4% to 88% with an average of 58%. Mean daily accumulative solar radiation in the greenhouse was 2.62 MJ/(m2·d), with a maximum of 5.41 MJ/(m2·d) at May 21st.
Treatments of root zone moisture level were applied to each pot after transplanting. Root zone moisture was indicated by the relative water content, the percent of volumetric water content comparing to field water capacity. Because of the dynamic change of root zone moisture status, upper limit was applied to define the moisture level of root zone. In other word, irrigation stopped when root zone moisture reached the upper limit. Considering the feasibility of moisture level control and plant growth, three root zone moisture levels were applied, 40%, 60%, and 80% (referred to as ‘Low’, ‘Medium’, and ‘High’ hereafter, respectively). In total, five plants of each cultivar were randomly assigned to each treatment. Irrigation was conducted manually for all plants once a day according to the pot weight.
Image acquisition and processing
A commercial phenotyping system (Scanalyzer3D, LemnaTec GmbH, Würselen, Germany) was used for image acquisition (Supplementary Fig. S2). Images of each plant were taken daily from top view before irrigation from May 11th to 29th, 2015. Two types of image were acquired simultaneously, near-infrared (NIR) and visible (VIS) images. The 20-day growing period was divided into five stages with four days of each, referred to as stage 1, 2, 3, 4, and 5. Plant growing level especially the number of leaves was different at different stage (Fig. 6). The reference growth stage was divided mainly according to the new leaf emerge time under different treatments and adjusted to even distribution for model comparison.
Two image acquisition schedules were applied with one from 06:00 h to 18:00 h at two hour intervals and the other one only at 14:00 h. Weather condition was classified into two scenarios, sunny and cloudy (referred to as ‘S’ and ‘C’ hereafter, respectively) according to weather forecast. Efforts were made to have the two image acquisition schedules for each weather scenario in each stage (Table 5). At the end of 20-day growing period, the fresh and dry weight of each plant sample were measured. Each sample was dried at 105 °C for 1 hour and then at 60 °C to constant weight. The dried sample was weighted to an accuracy of ± 0.1 mg.
Images acquired by the imaging system were organized into LemnaBase which is the central database interface for the phenotyping system, and processed through an analysis pipeline specifically adjusted for pakchoi by using LemnaGrid which is an image analysis component of the phenotyping system. Image processing procedure in LemnaGrid consisted of four main steps shown as the flowchart in Supplementary Fig. S3: (1) image preprocessing, extracting target images from LemnaBase; (2) segmentation, separating target plant from the background in the image; (3) feature extraction, analyzing segmentation result and producing phenotypic traits; and (4) post-processing, summarizing feature extraction results of all target images and exporting as “.xls” file.
Root zone water status discrimination model development
The development of root zone water status discrimination was consisted of three steps: (1) data preprocessing, organizing data for phenotypic analysis and model development; (2) phenotypic traits screening, removing those phenotypic traits with insignificant difference between treatments of root zone moisture level; (3) model development, using different modeling algorithms to classify root zone water status. MetaboAnalyst statistical analysis module (http://www.metaboanalyst.ca) was implemented for data preprocessing and phenotypic traits screening, and R language (windows, R3.2.5, R core Team, 2016) was implemented for model development.
In MetaboAnalyst statistical analysis module, data of extracted phenotypic traits from each image were organized into one datasheet by sample ID (row) and phenotypic traits (column). Empty rows were detected and excluded after data uploaded. Columns with more than 50% empty records were removed. In the case of columns with data missing but less than 50%, the NAs (not available values) were replaced by with a small value (half of the minimum positive values in the original data). All the traits value were normalized by using auto scaling method (mean-centered and divided by the standard deviation of each variable).
Phenotypic traits screening
Among those phenotypic traits extracted by imaging processing, traits which are insignificantly relevant to treatments may influence the modeling accuracy. In this study, one-way Analysis of Variance (ANOVA) was applied to screen traits resulting an optimal set of explanatory variables for discrimination model development. Traits with P > 0.05 (Fisher’s LSD method) were excluded from the traits set. In addition, the result of ANOVA presented a preliminary overview of the significance of each trait to treatments.
In order to assess the potentiality of using phenotypic traits to develop discrimination model, Partial Least Squares Discriminant Analysis (PLS-DA) was applied, which is a supervised algorithm that uses multivariate regression techniques to predict class membership via linear combination of original variables. The PLS-DA was performed by using plsr function in pls package of R language, and the classification and cross-validation were performed by using the corresponding wrapper function in caret package of R language. The resulted PLS-DA three-dimensional component score plot, which shows the overall data distribution of different treatments, presents the preliminary classification indicating modeling potentiality.
Discrimination model development
Based on the purpose of multi-classification in this study, three classification modeling algorithms, Random Forests (RF), Neural Network (NN), and Support Vector Machine (SVM) were selected to develop discrimination model and performed with the randomForest, nnet, e1071 packages in R language, respectively.
RF was used as a supervised learning algorithm suitable for the case of high dimensional data analysis. The RF can handle multiple input variables from large databases without variable selection and give estimates of important variables in the classification25. RF modeling was performed by using the randomForest package. Ntree (the number of trees) and mtry (the number of features) used to find the best feature were required in RF. In this study the RF model was trained with ntree as 500 and mtry from 2, 5, and 8.
Neural network have been successful as predictive tools in variety domains such as uncovering genotype-phenotype interactions26. Researchers reported that properly trained deep neural networks could discover, model, and disentangle latent factors for phenotyping27. For NN model in this study, nnet package was used to train the single-hidden-layer feed-forward neural network. There are two main parameters, decay and size, which can be tuned for NN model in R. In this study, the decay were tuned from 0, 0.1, and 0.001, and the size were tuned from 2, 5, and 9.
The goal of a SVM is to create a boundary, called hyperplane, which leads to homogeneous partitions of data on either side. In this study, multiclass SVM model were trained with kernel parameter C (from 10, 100, and 1000) and regularization parameter gamma (from 0.001, 0.01, 0.1, and 0). The best combination of C and gamma leading to the highest prediction accuracy was chosen. For SVM, svm function in e1071 package was used to specify the kernel function, cost, and the gamma function.
The 80% of total samples were selected as the training dastaset via stratified random sampling method for model development and optimization and the rest 20% as the test dataset. In order to prevent the over fitting problems, all three models were trained with the 10-fold cross validation in three repetitions by using the caret package in R language. This method randomly splits the dataset in 10 subsets, among which 9 instances of the data were used to train the model and the other instance was used to validate the training model. By performing three repeats of the 10-fold validation, the average accuracy (percentage of correctly classified samples out of all samples) was used to assess the performance of the model. However, only accuracy is not enough to evaluate the model performance especially for imbalance data or multi class data. Therefore, Cohen’s Kappa coefficient (k) and ROC (Receiver operating characteristic) curve were also used to evaluate the model performance. The k is a good measure that can handle multi-class well and calculated as:
where, po is the observed agreement, and pe is the expected agreement. The ROC curves of different models were compared according to the area under the curve (AUC) by plotting the true positive rate against the false positive rate. AUC represents the probability that the classifier will assign a higher score to a randomly chosen positive example than to negative example28.
Publisher's note: Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
This study was supported in part by the National Natural Science Foundation of China (NSFC) (Project No. 31601214), National High-tech R&D Program of China (863 Program, Project No. 2013AA103006), Science and Technology Commission of Shanghai Municipality (Project No. 16391901700), and Shanghai Jiao Tong University Agri-X Foundation. Any opinions, findings, conclusions, or recommendations expressed in this publication are those of the authors and do not necessarily reflect the view of the National Natural Science Foundation of China, Ministry of Science and Technology of China, Science and Technology Commission of Shanghai Municipality, or Shanghai Jiao Tong University.
Electronic supplementary material
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/.