当特征用稀疏矩阵表示时,我的 SVM 分类器会抛出值错误,但如果特征用密集矩阵表示则不会出错。
我有对我的功能集执行 One Hot Encoding 的代码,并将编码的输出添加到新的功能列表中。当使用 .toarray() 将 One Hot Encoding 的输出转换为密集数组时,我的 SVM 分类器运行良好。
但是,使用密集数组并不理想,因为我有数千个数据点,而且我的计算机很快就会耗尽内存。因此,需要稀疏数组。如果我只是从下面的代码中删除 .toarray() ,则 enc.transform(features) 的输出将输出一个稀疏矩阵。但是,如果我运行我的 SVM 分类器,我现在会收到以下错误:
ValueError:使用序列设置数组元素。
当我的 SVM 尝试拟合数据时,似乎有些事情失败了。Sklearn SVM 接受稀疏向量,所以我不明白出了什么问题。
# Perform One Hot Encoding
transformedFeatureList = []
for features in featureList:
features = np.asarray(features)
features = features.reshape(1, -1)
transformedFeatures = enc.transform(features).toarray() <---Without toarray() the Value Error happens
transformedFeatureList.append(transformedFeatures)
featureList = transformedFeatureList
# Seperate data into training and testing set
trainingSet = [[], []]
testSet = [[], []]
if len(featureList) == len(classList):
for index in range(len(featureList)):
if random.randint(1, 10) <= 7:
trainingSet[0].append(featureList[index])
trainingSet[1].append(classList[index])
else:
testSet[0].append(featureList[index])
testSet[1].append(classList[index])
# Train model and attempt classification
from sklearn import svm
X = trainingSet[0]
y = trainingSet[1]
clf = svm.SVC()
clf.fit(X, y)
results = {}
for iclass in set(classList):
results[iclass] = [0, 0] # index 0 = correct, index 1 = incorrect
if len(testSet[0]) == len(testSet[1]):
for index in range(len(testSet[0])):
features = testSet[0][index]
iclass = testSet[1][index]
predictedClass = clf.predict([features])[0]
if predictedClass == iclass:
results[iclass][0] += 1
else:
results[iclass][1] += 1