add plot in machine learning