Scikit Learn Numpy Python Code