Transformers for wastewater treatment

The project explores transformer based models for simulating removal of heavy metals from wastewater using adsorption proecss. We will use two transformer based models i.e. Tab-Transformer and FT-Transformer. Other than using transformers for wastewater efficiency prediction, We will perform the following tasks

• how to perform feature selection for FT-Transformer model

• Compare different validation schemes for model selection

• Impact of logrithmic transformation on model prediction

The dataset initially consisted of 1514 samples and 28 features.

Scripts

Notebooks

• Recursive Feature Elimination
• Held out validation
• KFold Cross Validation
• Grouped KFold