Machine learning steps - Python programming