Fixed the confusion matrix.

This commit is contained in:
Batuhan Berk Başoğlu 2025-09-29 23:06:53 -04:00
parent be12360f9a
commit 193dcabbff
Signed by: batuhan-basoglu
SSH key fingerprint: SHA256:kEsnuHX+qbwhxSAXPUQ4ox535wFHu/hIRaa53FzxRpo
2 changed files with 4 additions and 4 deletions

View file

@ -1,7 +1,7 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from matplotlib import pyplot as plt 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: class LogisticRegression:
@ -131,7 +131,7 @@ class LogisticRegression:
y_pred = self.predict(x) y_pred = self.predict(x)
y_true = np.asarray(y).astype(int) 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: if normalize:
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis] cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]

View file

@ -1,7 +1,7 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from matplotlib import pyplot as plt 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: class LogisticRegression:
''' '''
@ -149,7 +149,7 @@ class LogisticRegression:
y_pred = self.predict(x) y_pred = self.predict(x)
y_true = np.asarray(y).astype(int) 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: if normalize:
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis] cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]