clean up and comment machine learning scripts