Coastal water quality prediction based on machine learning with feature interpretation and spatio-temporal analysis

Grbčić, Luka; Družeta, Siniša; Mauša, Goran; Lipić, Tomislav; Lušić, Darija Vukić; Alvir, Marta; Lučin, Ivana; Sikirica, Ante; Davidović, Davor; Travaš, Vanja; Kalafatovic, Daniela; Pikelj, Kristina; Fajković, Hana; Holjević, Toni; Kranjčević, Lado (2022) Coastal water quality prediction based on machine learning with feature interpretation and spatio-temporal analysis. Environmental Modelling & Software, 155 . ISSN 1364-8152

PDF - Submitted Version - article Available under License Creative Commons Attribution Non-commercial No Derivatives. Download (4MB)

Abstract

Coastal water quality management is a public health concern, as water of poor quality can potentially harbor dangerous pathogens. In this study, we employ routine monitoring data of Escherichia Coli and enterococci across 15 beaches in the city of Rijeka, Croatia, to build machine learning models for predicting E. Coli and enterococci based on environmental features. Cross-validation analysis showed that the Catboost algorithm performed best with R values of 0.71 and 0.69 for predicting E. Coli and enterococci, respectively, compared to other evaluated algorithms. SHapley Additive exPlanations technique showed that salinity is the most important feature for forecasting both E. Coli and enterococci levels. Furthermore, for low water quality sites, the spatial predictive models achieved R values of 0.85 and 0.83, while the temporal models achieved R values of 0.74 and 0.67. The temporal model achieved moderate R^2 values of 0.44 and 0.46 at a site with high water quality.

Item Type:	Article
Uncontrolled Keywords:	coastal water quality ; machine learning ; shap ; catboost ; fecal indicator bacteria
Subjects:	TECHNICAL SCIENCES > Computing BIOTECHNICAL SCIENCES > Interdisciplinary Technical Sciences
Divisions:	Center for Informatics and Computing Division of Electronics
Projects:	Project title Project leader Project code Project type Računalni model strujanja, poplavljivanja i širenja onečišćenja u rijekama i obalnim morskim područjima-KLIMOD Lado Kranjčević; Darija Vukić Lušić; Davor Davidović KK.05.1.1.02.0017 EK
Depositing User:	Ana Zečević
Date Deposited:	07 Apr 2026 06:18
URI:	http://fulir.irb.hr/id/eprint/11579
DOI:	10.1016/j.envsoft.2022.105458

Actions (login required)

View Item