Comprehensive assessment of the water environment carrying capacity based on machine learning