The objective of this study was to consider the efficiency of support vector machine (SVM) and artificial neural network (ANN) for the classification and prediction of groundwater quality using a small data record in Malayer, Iran. For this purpose, 14 groundwater quality variables that had been collected from 27 groundwater sampling wells were used. Cluster analysis discriminated the total sampling stations into two groups. The classification was implemented by SVM using polynomial and RBF kernel methods. The respective sensitivity and specificity of this model were 0.89 and 0.80 while that of positive predictive value and negative predictive value were 0.89 and 0.86, respectively. The prediction of water quality index (WQI) was implemented using ANN. Despite the high correlation coefficient between the predicted and observed values of WQI(r = 0.90), the generalization ability of this model was low(r = 0.60) indicating the over-fitting of the model to the training data set.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.