Machine Learning trial with python scikit-learn package