TODO: (1) fix partial_fit and fit in LogisticRegression (2) Finish LogisticRegression cg