The application of machine learning methods for prediction of heavy metal by activated carbons, biochars, and carbon nanotubes

Carbonaceous materials are commonly used as adsorbents for heavy metals. The determination of the adsorption capacity needs time and energy, and the key factors affecting the adsorption capacity have not been determined. Therefore, a new and efficient method is needed to predict the adsorption capacity and explore the decisive factors in the adsorption process. In this study, three tree-based machine learning models (i.e., random forest, gradient boosting decision tree, and extreme gradient boosting) were developed to predict the adsorption capacity of eight heavy metals (i.e., As, Cd, Cr, Cu, Hg, Ni, Pb, and Zn) on activated carbons, biochars, and carbon nanotubes using 3674 data points extracted from 151 journal articles. After a comprehensive comparison, the gradient boosting decision tree had the best performance for a combined model based on all data (R2 = 0.9707, RMSE = 0.1420). Moreover, independent models were developed for three datasets classified by the adsorbent and eight datasets classified by the heavy metals. In addition, a graphical user interface was built to predict the adsorption capacity of heavy metals. This study provides a novel strategy and convenient tool for the removal of heavy metals and can help to improve the removal efficiency of heavy metals to build a healthier world.