machine learning code and comments