重塑支持向量机的ndarray

时间:2016-03-21 08:09:13

标签: python numpy machine-learning svm

我想使用SVHN数据并尝试使用SVM。

testdata['X'] <type 'numpy.ndarray'>


(testdata['X']).shape is (32, 32, 3, 26032)
问题是SVM需要一个2d数组,而我的是4。 这意味着我需要重新塑造它。

我试过了:

(testdata['X']).reshape(2)

给了我:

ValueError: total size of new array must be unchanged

1 个答案:

答案 0 :(得分:1)

使用重塑时,您需要使用新数组中前一个数组中的所有元素,例如,如果您的尺寸为:

(testdata['X']).shape is (x1, x2, x3, x4)

你可以用这种方式重塑:

(testdata['X']).reshape(x1*x2*x3,x4)
根据您的需要

或其他一些组合