Wastewater discharge quality prediction using stratified sampling and wavelet de-noising ANFIS model

•A general input parameter selection method is proposed, combined with an optimized wavelet de-noising ANFIS model to predict wastewater quality.•A network structure selection method is given to enhance the stability and scalability of ANFIS.•Statistical stratified sampling is employed to proportionally partition the dataset.•The experimental results confirm that the proposed model outperforms other existing AI models.