From 8f48f17b4ed58326a06a223813d02ee51ecd64b7 Mon Sep 17 00:00:00 2001
From: ukfnu <ukfnu@student.kit.edu>
Date: Thu, 30 Jan 2025 20:40:16 +0100
Subject: [PATCH] added logistic regression
---
code/src/machine_learning.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/code/src/machine_learning.py b/code/src/machine_learning.py
index 8aedf99..83c07bd 100644
--- a/code/src/machine_learning.py
+++ b/code/src/machine_learning.py
@@ -24,6 +24,7 @@ from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import MinMaxScaler
from sklearn.naive_bayes import GaussianNB
+from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score
try:
@@ -98,6 +99,7 @@ clfg = GaussianNB()
start = time.time()
clfg.fit(x_train, y_train.values.ravel())
end = time.time()
+print("Naive Bayes")
print(f"Training time: {end - start}")
start = time.time()
@@ -106,3 +108,19 @@ end = time.time()
print(f"Prediction time: {end - start}")
accuracy = accuracy_score(y_test, y_pred)
print(f"Accuracy: {accuracy}")
+
+
+# Logistic Regression
+print("\n\nLogistic Regression")
+reg = LogisticRegression()
+start = time.time()
+reg.fit(x_train, y_train.values.ravel())
+end = time.time()
+print(f"Training time: {end - start}")
+
+start = time.time()
+y_pred = reg.predict(x_test)
+end = time.time()
+print(f"Prediction time: {end - start}")
+accuracy = accuracy_score(y_test, y_pred)
+print(f"Accuracy: {accuracy}")
\ No newline at end of file
--
GitLab