scikit-learn – 使用svm.svc分类器进行多标签分类,是否可以在不设置probability=True的情况下实现?

我尝试使用scikit-learn中的Pipeline和OneVsRest分类器来实现多标签分类。代码如下,但首先我要提到,我是从pandas数据框中构建我的多标签示例的。

代码如下:

df = pd.read_csv(fileIn, header = 0, encoding='utf-8-sig')rows = random.sample(df.index, int(len(df) * 0.9))work = df.ix[rows]work_test = df.drop(rows)X_train = []y_train = []X_test = []y_test = []for i in work[[i for i in list(work.columns.values) if i.startswith('Change')]].values:    X_train.append(','.join(i.T.tolist()))X_train = np.array(X_train)for i in work[[i for i in list(work.columns.values) if i.startswith('Corax')]].values:    y_train.append(list(i))for i in work_test[[i for i in list(work_test.columns.values) if i.startswith('Change')]].values:    X_test.append(','.join(i.T.tolist()))X_test = np.array(X_test)for i in work_test[[i for i in list(work_test.columns.values) if i.startswith('Corax')]].values:    y_test.append(list(i))lb = preprocessing.MultiLabelBinarizer()Y = lb.fit_transform(y_train)classifier = Pipeline([('vectorizer', CountVectorizer()),('tfidf', TfidfTransformer()),('clf', OneVsRestClassifier(SVC(kernel='rbf')))])classifier.fit(X_train, Y)predicted = classifier.predict(X_test)

但问题是,当你使用这一系列转换:CountVectorizer -> TfidfTransformer时,你会得到一个稀疏矩阵。问题在于,当你尝试使用OneVsRest分类器预测标签时,它会寻找decision_functionpredict_proba方法。在svm.SVC中,除非你指定probability=True,否则predict_proba是不可用的。另一方面,从代码中可以看出,decision_function对于稀疏矩阵并未实现。因此我的代码失败了,因为这两个必需的方法都不可用。但也许我做错了什么?是否有可能在不指定probability=True的情况下使用svm.SVC实现多标签分类?(这样做会显著增加分类器训练的开销),也许可以通过强制TfidfTransformer输出密集矩阵而不是稀疏矩阵来实现?


回答:

这是一个众所周知的问题,目前还没有简单的解决方案。

你可以使用Pipeline来“密集化”你的稀疏数据(通过调用.toarray),但这可能会大幅增加内存消耗。你可以使用TruncatedSVD(据我所知,这是唯一适用于稀疏数据的降维方法),但这可能会干扰你的数据,从而降低SVM的性能。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注