logistic_regression.py

def create_logistic_regressions(X_train,y_train,figsize=(10,10)):
  logreg = LogisticRegression(solver='lbfgs')
  logreg.fit(X_train, y_train)

  coefficients = logreg.coef_
  intercept = logreg.intercept_
  
  df_logreg = pd.DataFrame({'Feature':X_train.columns,'Coef':logreg.coef_[0]})
  fig, ax = plt.subplots(figsize=figsize)
  sns.barplot(x="Coef", y="Feature", data=df_logreg, ax=ax)
  return logreg, df_logreg.set_index('Feature')