https://github.com/aneagoie/ML-Notes/blob/master/iris.ipynb