go through sentdex machine learning tutorial https://pythonprogramming.net/machine-learning-tutorial-python-introduction/