ML classification model (Python) | Shghilni