From e6be505c04706b33aaac4f6ae44e88203e1b2988 Mon Sep 17 00:00:00 2001 From: lbecker <lbecker@rhrk.uni-kl.de> Date: Sun, 18 Aug 2024 06:02:30 +0200 Subject: [PATCH] Upload New File --- logreg_builder.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 logreg_builder.py diff --git a/logreg_builder.py b/logreg_builder.py new file mode 100644 index 0000000..ef69f5b --- /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 -- GitLab