Prediction of River Water Quality Based on Artificial Neural Network

Prediction of River Water Quality Based on Artificial Neural Network


  • Danial Mustaqim Azmi College of Computing, Informatics and Mathematics, Universiti Teknologi MARA Terengganu Branch, Kuala Terengganu Campus, Terengganu, Malaysia
  • Norlina Mohd Sabri College of Computing, Informatics and Mathematics, Universiti Teknologi MARA Terengganu Branch, Kuala Terengganu Campus, Terengganu, Malaysia
  • Nik Marsyahariani Nik Daud College of Computing, Informatics and Mathematics, Universiti Teknologi MARA Terengganu Branch, Kuala Terengganu Campus, Terengganu, Malaysia
  • Nor Azila Awang Abu Bakar College of Computing, Informatics and Mathematics, Universiti Teknologi MARA Terengganu Branch, Kuala Terengganu Campus, Terengganu, Malaysia


Water Analysis, River Quality, Prediction, Artificial neural network


In machine learning, prediction is a method that is supported by historical data and is often used in various fields. It can be used to predict quality for water taken from river, which is a major source of life particularly for human. Water contamination may be a result of civilization and the rapid increment in economy. This research looks at how the Artificial Neural Network (ANN) algorithm predicts the Water Quality Index that will aid the environmental agencies and consumers. This study also aims to create an ANN-based water quality index prediction system that is effective at managing noisy data. Furthermore, the constructed system’s ability to accurately predict water quality is assessed. The water quality prediction method is based on the data taken from the three main rivers in Selangor, namely Sungai Buloh, Langat, and Kuala Selangor. The prediction takes into account variables such as Biological Oxygen Demand (BOD), dissolve oxygen (DO) and seven other water characteristics. The performance metric used in the study is the calculation of the accuracy for factors such as the number of neurons in the hidden layer, the epoch number, the split data ratio and the learning rate. The result has shown that the ANN model has produced good and acceptable performance with 88.44% accuracy. For future work, the ANN model can be improved by collecting more data for its training and the performance of the model can be compared with other prediction algorithms.


Download data is not yet available.


Afandi, A., Lusi, N., Catrawedarma, I., & Rudiyanto, B. (2021). Prediction of temperature in 2 meters temperature probe survey in Blawan Geothermal Field using artificial neural network (ANN) method. SSRN Electronic Journal.

Aldrees, A., Awan, H. H., Javed, M. F., & Mohamed, A. M. (2022, December). Prediction of water quality indexes with ensemble learners: Bagging and boosting. Process Safety and Environmental Protection, 168, 344–361.

Azimi, S., Azhdary Moghaddam, M., & Hashemi Monfared, S. A. (2019). Prediction of annual drinking water quality reduction based on Groundwater Resource Index using the artificial neural network and fuzzy clustering. Journal of Contaminant Hydrology, 220, 6–17.

Babu, S., Nagaleela, B. B., Karthik, C.G., Yepuri, L.N. (2023). Water Quality Prediction using Neural Networks. 2023 International Conference on Artificial Intelligence and Knowledge Discovery in Concurrent Engineering (ICECONF) (pp. 1-6). IEEE Xplore. doi: 10.1109/ICECONF57129.2023.10084120.

Bi, J., Lin, Y., Dong, Q., Yuan, H., & Zhou, M. (2021, September). Large- scale water quality prediction with integrated deep neural network. Information Sciences, 571, 191–205.

Dawood, T., Elwakil, E., Novoa, H. M., & Gárate Delgado, J. F. (2021). Toward urban sustainability and clean potable water: Prediction of water quality via artificial neural networks. Journal of Cleaner Production, 291.

Ghosh, A., Mukherjee, J., & Chakravorty, N. (2023, February). A low-cost test for anemia using an Artificial Neural Network. Computer Methods and Programs in Biomedicine, 229, 107251.

Guo, L. And Fu, D. ((2023). River water quality prediction model based on PCA-APSO-ELM neural network. 2023 6th International Conference on Artificial Intelligence and Big Data (ICAIBD). Chengdu, China (pp. 512-517). IEEE Xplore. doi: 10.1109/ICAIBD57115.2023.10206249.

Ho, J. Y., Afan, H. A., El-Shafie, A. H., Koting, S. B., Mohd, N. S., Jaafar, W. Z. B., Lai Sai, H., Malek, M. A., Ahmed, A. N., Mohtar, W. H. M. W., Elshorbagy, A., & El-Shafie, A. (2019). Towards a time and cost effective approach to water quality index class prediction. Journal of Hydrology, 575, 148–165.

Kim, J., Seo, D., Jang, M., & Kim, J. (2021, November). Augmentation of limited input data using an artificial neural network method to improve the accuracy of water quality modeling in a large lake. Journal of Hydrology, 602, 126817.

Leong, W. C., Bahadori, A., Zhang, J., & Ahmad, Z. (2019, June 19). Prediction of water quality index (WQI) using support vector machine (SVM) and least square-support vector machine (LS-SVM). International Journal of River Basin Management, 19(2), 149–156.

Lu, F., Liang, Y., Wang, X., Gao, T., Chen, Q., Liu, Y., Zhou, Y., Yuan, Y., & Liu, Y. (2022, July). Prediction of amorphous forming ability based on artificial neural network and convolutional neural network. Computational Materials Science, 210, 111464.

Murivhami, T., Tartibu, L.K. and Olayode, I.O (2023). Applications of artificial neural network in the prediction of water quality index. 2023 14th International Conference on Mechanical and Intelligent Manufacturing Technologies (ICMIMT) (pp. 100-104). IEEE Xplore. doi: 10.1109/ICMIMT59138.2023.10199388.

Najah Ahmed, A., Binti Othman, F., Abdulmohsin Afan, H., Khaleel Ibrahim, R., Ming Fai, C., Shabbir Hossain, M., Ehteram, M., & Elshafie, A. (2019, November). Machine learning methods for better water quality prediction. Journal of Hydrology, 578, 124084.

Nayak, J. G., Patil, L., & Patki, V. K. (2023). Artificial neural network based water quality index (WQI) for river Godavari (India). Materials Today: Proceedings, 81, 212–220.

Olabi, A., Abdelkareem, M. A., Semeraro, C., Radi, M. A., Rezk, H., Muhaisen, O., Al-Isawi, O. A., & Sayed, E. T. (2023, January). Artificial neural networks applications in partially shaded PV systems. Thermal Science and Engineering Progress, 37, 101612.

Ozcalici, M., & Bumin, M. (2022, December). Optimizing filter rule parameters with genetic algorithm and stock selection with artificial neural networks for an improved trading: The case of Borsa Istanbul. Expert Systems with Applications, 208, 118120.

Pantic, I., Paunovic, J., Cumic, J., Valjarevic, S., Petroianu, G. A., & Corridon, P. R. (2023, January). Artificial neural networks in contemporary toxicology research. Chemico-Biological Interactions, 369, 110269.

Portillo Juan, N., & Negro Valdecantos, V. (2022, September). Review of the application of artificial neural networks in ocean engineering. Ocean Engineering, 259, 111947.

Satish, N., Anmala, J., Rajitha, K., Varma, M. R. R., (2024). A stacking ANN ensemble model of ML models for stream water quality prediction of Godavari River Basin, India. Ecological Informatics, 80, 102500.

Sattari, M. T., Feizi, H., Colak, M. S., Ozturk, A., Ozturk, F., & Apaydin, H. (2021). Surface water quality classification using data mining approaches: Irrigation along the Aladag River*. Irrigation and Drainage, 70(5), 1227–1246.

Xia, L., Han, Q., Shang, L., Wang, Y., Li, X., Zhang, J., Yang, T., Liu, J., & Liu, L. (2022). Quality assessment and prediction of municipal drinking water using water quality index and artificial neural network: A case study of Wuhan, central China, from 2013 to 2019. Science of the Total Environment, 844.

Ying Ying Tang, D., Wayne Chew, K., Ting, H. Y., Sia, Y. H., Gentili, F. G., Park, Y. K., Banat, F., Culaba, A. B., Ma, Z., & Loke Show, P. (2023, February). Application of regression and artificial neural network analysis of Red-Green-Blue image components in prediction of chlorophyll content in microalgae. Bioresource Technology, 370, 128503.

Zamri, N., Pairan, M. A., Azman, W. N. A. W., Abas, S. S., Abdullah, L., Naim, S., Tarmudi, Z., & Gao, M. (2022). River quality classification using different distances in k-nearest neighbors algorithm. Procedia Computer Science, 204, 180–186.




How to Cite

Azmi, D. M., Mohd Sabri, N., Nik Daud, N. M., & Awang Abu Bakar, N. A. (2024). Prediction of River Water Quality Based on Artificial Neural Network. Journal of Computing Research and Innovation, 9(1), 91–106. Retrieved from



General Computing

Similar Articles

You may also start an advanced similarity search for this article.