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

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

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