Leetcode Machine Learning Python