batuhan-basoglu/Mini-Project-of-Machine-Learning
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fixed the confusion matrix.

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
mini-batch-sgd-logistic-regression-wdbc.py
View file

@ -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]