https://pylessons.com/Logistic-Regression-part3