python调用数据集_在sklearn中使用数据集

您需要有两个不同的标签来绘制ROC曲线。

如果我在您的数据中添加一些0标签，下面的示例适用于我。我用熊猫来读取数据，其余都是和sklearn一样的例子。在

此外，您需要将数据集拆分为训练集和测试集，以便在测试集上绘制ROC曲线。在import pandas as pd

import numpy as np

from scipy import interp

import pylab as pl

from sklearn import svm

from sklearn.metrics import roc_curve, auc

from sklearn.cross_validation import StratifiedKFold

def data(filename):

X = pd.read_table(filename, sep=',', warn_bad_lines=True, error_bad_lines=True, low_memory = False)

X = np.asarray(X)

data = X[:,2:]

labels = X[:,1]

print np.unique(labels)

return data, labels

filename = '../data/sodata.csv'

X, y = data(filename)

###############################################################################

# Classification and ROC analysis

#