Python Machine Learning Scatter Plot