使用sklearn用chi平方核预测多标记

时间:2015-03-13 05:17:31

标签: python machine-learning scikit-learn svm

我正在尝试使用预先计算的卡方内核来获得SVM的预测。但是,在尝试运行clf.predict()时遇到问题。

min_max_scaler = preprocessing.MinMaxScaler()
X_train_scaled = min_max_scaler.fit_transform(features_train)
X_test_scaled = min_max_scaler.transform(features_test)

K = chi2_kernel(X_train_scaled)
svm = SVC(kernel='precomputed', cache_size=1000).fit(K, labels_train)
y_pred_chi2 = svm.predict(X_test_scaled)

我得到的错误如下:

ValueError: bad input shape (4627L, 20L)

我猜这个问题是因为多标签,所以我通过以下方式训练分类器只有1个类别:

svm = SVC(kernel='precomputed', cache_size=1000).fit(K, labels_train[:, 0])

但是,在尝试运行clf.predict(X_test_scaled)时,我收到错误:

ValueError: X.shape[1] = 44604 should be equal to 4627, the number of samples at training time

为什么测试样本必须与训练样本的数量相同?

以下是相关矩阵的形状(这些特征有44604个维度,有20个类别):

X_train_scaled.shape    : (4627L, 44604L)
X_test_scaled.shape     : (4637L, 44604L)
K.shape                 : (4627L, 4627L)
labels_train.shape      : (4627L, 20L)

另一方面,这些矩阵的形状大小旁边是否正常?

2 个答案:

答案 0 :(得分:1)

您需要在测试数据和训练数据之间为内核提供预测函数。最简单的方法是给内核参数kernel=chi2_kernel一个可调用的。 使用

K_test = chi2_kernel(X_test_scaled)

无效。需要

K_test = chi2_kernel(X_test_scaled, X_train_scaled)

答案 1 :(得分:0)

clf.predict()的输入也必须传递给chi2_kernel函数。

K_test = chi2_kernel(X_test_scaled)
y_pred = svm.predict(K_test)