Вот пример моего набора данных:
Pat_ID Flare_Up Demo1 Demo2 Demo3 Demo4 Demo5 Demo6 DisHis1 DisHis1Times DisHis2 ... Dis6Treat Dis7 RespQues1 ResQues1a ResQues1b ResQues1c ResQues2a SmokHis1 SmokHis2 SmokHis3 SmokHis4
0 1 0 1 0.246004 0.391931 0.237792 0 0.443526 0 0.000000 0 ... 1 0 0.12623 0.1032 0.2439 0.0597 0.0 0.411765 0.263620 0.482759 0.1875
1 2 1 1 0.225851 0.268012 0.268481 0 0.286501 0 0.000000 1 ... 1 0 0.60707 0.3808 0.8637 0.4949 0.1 0.117647 0.098418 0.624138 0.0000
2 3 0 0 0.342599 0.476945 0.296468 1 0.159780 1 0.166667 1 ... 0 0 0.77541 0.6318 1.0000 0.6570 0.3 0.035294 0.020211 0.510345 0.0000
[3 rows x 62 columns]
Мой код для обхода этого набора данных и печати ROC
:
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.linear_model import LogisticRegression
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import StratifiedKFold
from sklearn.metrics import confusion_matrix, classification_report, roc_curve, roc_auc_score
import itertools
def plot_confusion_matrix(cm, classes, normalize=True, title='Confusion matrix', cmap=plt.cm.Blues):
"""
This function prints and plots the confusion matrix.
Normalization can be applied by setting `normalize=True`.
"""
plt.imshow(cm, interpolation='nearest', cmap=cmap)
plt.title(title)
plt.colorbar()
tick_marks = np.arange(len(classes))
plt.xticks(tick_marks, classes, rotation=45)
plt.yticks(tick_marks, classes)
if normalize:
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
print("Normalized confusion matrix")
# else:
# print('Confusion matrix, without normalization')
# print(cm)
thresh = cm.max() / 2.
for i, j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):
plt.text(j, i, cm[i, j],
horizontalalignment="center",
color="white" if cm[i, j] > thresh else "black")
plt.tight_layout()
plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()
def show_data(cm, print_res = 0):
tp = cm[1,1]
fn = cm[1,0]
fp = cm[0,1]
tn = cm[0,0]
if print_res == 1:
print('Precision = {:.3f}'.format(tp/(tp+fp)))
print('Recall (TPR) = {:.3f}'.format(tp/(tp+fn)))
print('Fallout (FPR) = {:.3e}'.format(fp/(fp+tn)))
return tp/(tp+fp), tp/(tp+fn), fp/(fp+tn)
df = pd.read_csv("datasource/DevelopmentData.csv")
print(df.head(3))
y = np.array(df.Class.tolist()) #classes: 1..fraud, 0..no fraud
df = df.drop('Class', 1)
df = df.drop('Time', 1) # optional
df['Amount'] = StandardScaler().fit_transform(df['Amount'].values.reshape(-1,1)) #optionally rescale non-normalized column
X = np.array(df.as_matrix()) # features
Класс 0 означаетчто транзакция была в порядке, а класс 1 означает, что транзакция была мошеннической.
Когда я запускаю свой код, я получаю эту ошибку:
Traceback (most recent call last):
File "finalindex.py", line 54, in <module>
y = np.array(df.Class.tolist()) #classes: 1..fraud, 0..no fraud
File "C:\Users\kulkaa\AppData\Local\Programs\Python\Python37-32\lib\site-packages\pandas\core\generic.py", line 4376, in __getattr__
return object.__getattribute__(self, name)
AttributeError: 'DataFrame' object has no attribute 'Class'
Как я могу исправить эту ошибку?Нужно ли менять имена столбцов в соответствии с набором данных?