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