From 193dcabbffe7bd2d8471d60afe9f0d8d14d2d4ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Batuhan=20Berk=20Ba=C5=9Fo=C4=9Flu?= Date: Mon, 29 Sep 2025 23:06:53 -0400 Subject: [PATCH] Fixed the confusion matrix. --- logistic-regression-wdbc.py | 4 ++-- mini-batch-sgd-logistic-regression-wdbc.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/logistic-regression-wdbc.py b/logistic-regression-wdbc.py index a6843ae..52d283d 100644 --- a/logistic-regression-wdbc.py +++ b/logistic-regression-wdbc.py @@ -1,7 +1,7 @@ import numpy as np import pandas as pd from matplotlib import pyplot as plt -from sklearn.metrics import confusion_matrix, roc_auc_score +from sklearn.metrics import confusion_matrix as sk_confusion_matrix, roc_auc_score class LogisticRegression: @@ -131,7 +131,7 @@ class LogisticRegression: y_pred = self.predict(x) y_true = np.asarray(y).astype(int) - cm = confusion_matrix(y_true, y_pred) + cm = sk_confusion_matrix(y_true, y_pred, labels=[0, 1]) if normalize: cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis] diff --git a/mini-batch-sgd-logistic-regression-wdbc.py b/mini-batch-sgd-logistic-regression-wdbc.py index 3009821..5941d59 100644 --- a/mini-batch-sgd-logistic-regression-wdbc.py +++ b/mini-batch-sgd-logistic-regression-wdbc.py @@ -1,7 +1,7 @@ import numpy as np import pandas as pd from matplotlib import pyplot as plt -from sklearn.metrics import confusion_matrix, roc_auc_score +from sklearn.metrics import confusion_matrix as sk_confusion_matrix, roc_auc_score class LogisticRegression: ''' @@ -149,7 +149,7 @@ class LogisticRegression: y_pred = self.predict(x) y_true = np.asarray(y).astype(int) - cm = confusion_matrix(y_true, y_pred) + cm = sk_confusion_matrix(y_true, y_pred, labels=[0, 1]) if normalize: cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]