Prediction of water quality index (WQI) using support vector machine (SVM) and least square-support vector machine (LS-SVM)

• Published in: International journal of river basin management Vol. 19; no. 2; pp. 149 - 156
• Main Authors: Leong, Wei Cong, Bahadori, Alireza, Zhang, Jie, Ahmad, Z.
• Format: Journal Article
• Language: English
• Published: Abingdon Taylor & Francis 03-04-2021; Taylor & Francis Ltd
• Subjects: Error analysis; Kernel functions; least square-support vector machine; Least squares; model prediction; Polynomials; Radial basis function; support vector machine; Support vector machines; Water quality; Water quality index
• ISSN: 1571-5124; 1814-2060
• DOI: 10.1080/15715124.2019.1628030

The current calculations of water quality index (WQI) were sometimes can be very complex and time-consuming which involves sub-index calculation like BOD and COD, however with the support vector machine (SVM) and least squares support vector machine (LS-SVM) models, the WQI can be predicted immediately using directly measured physical data by using the same predictors used in the numerical approach without any sub-index calculation. There were three main parameters that control the performance of the SVM model however only the type of kernel function was investigated, they were linear, radial basis function (RBF) and polynomial kernel functions. The results of the model were then analysed by using sum squares error (SSE), mean of sum squares error (MSSE) and coefficient of determination (R
2
). It was found that the best kernel function for the SVM model was polynomial kernel function with R
2
of 0.8796. Furthermore, the LS-SVM model that trained with correct predictors had higher accuracy with R
2
of 0.9227 as compared with SVM model that trained with all the predictors with R
2
of 0.9184. The SSE and MSSE are 74.78 and 1.5594, 1.6454 for LS-SVM and SVM respectively.