Python Machine Learning Game