Fixed the confusion matrix.
This commit is contained in:
parent
be12360f9a
commit
193dcabbff
2 changed files with 4 additions and 4 deletions
|
|
@ -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]
|
||||||
|
|
|
||||||
|
|
@ -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]
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue