Logistic Regression Python Code With A Main