diff --git a/logreg_builder.py b/logreg_builder.py new file mode 100644 index 0000000000000000000000000000000000000000..ef69f5b859f2880460d06a5289523229085e1cec --- /dev/null +++ b/logreg_builder.py @@ -0,0 +1,14 @@ +from clf_builder import clf_builder +from sklearn.linear_model import LogisticRegression + + +class logreg_builder(clf_builder): + def __init__(self,x_train,y_train,x_test,y_test,cv,params,scoring='accuracy'): + super().__init__(x_train,y_train,x_test,y_test,cv,params,scoring) + + def make_logreg(self): + base_log = LogisticRegression() + best_log,result_table = self.hyperparameter_grid_search('Logistic Regression',base_log) + base_log = base_log.fit(self.x_train,self.y_train) + train_test_base_best=self.compare_models('Logistic Regression', base_log, best_log) + return best_log,result_table,train_test_base_best \ No newline at end of file