diff --git a/code/src/machine_learning.py b/code/src/machine_learning.py index 8aedf996f8d8f46d56ef4fb2f7b9a6b6a9819f71..83c07bd067d780e31eb4cd751eb6365101dab490 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