Train the model on cleaned data