2016-03-20 3 views
1

Я использую sklearn для применения svm на моем собственном наборе изображений. Изображения помещаются в кадр данных. Я перехожу к функции соответствия массиву numpy, который имеет 2D-списки, эти 2D-списки представляют изображения, а второй вход, который я передаю функции, - это список целей (Цели - это числа). Я всегда получаю эту ошибку «ValueError: установка элемента массива с последовательностью».sklearn SVM fit() «ValueError: установка элемента массива с последовательностью»

trainingImages = images.ix[images.partID <=9] 
trainingTargets = images.clustNo.ix[images.partID<=9] 
trainingImages.reset_index(inplace=True,drop=True) 
trainingTargets.reset_index(inplace=True,drop=True) 

classifier = svm.SVC(gamma=0.001) 
classifier.fit(trainingImages.image.values,trainingTargets.values.tolist()) 

Ошибка:

--------------------------------------------------------------------------- 
ValueError        Traceback (most recent call last) 
<ipython-input-43-5336fbeca868> in <module>() 
     8 classifier = svm.SVC(gamma=0.001) 
     9 
---> 10 classifier.fit(trainingImages.image.values,trainingTargets.values.tolist()) 
    11 
    12 #classifier.fit(t, list(range(0,2899))) 

/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/sklearn/svm/base.py in fit(self, X, y, sample_weight) 
    148   self._sparse = sparse and not callable(self.kernel) 
    149 
--> 150   X = check_array(X, accept_sparse='csr', dtype=np.float64, order='C') 
    151   y = self._validate_targets(y) 
    152 

/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/sklearn/utils/validation.py in check_array(array, accept_sparse, dtype, order, copy, force_all_finite, ensure_2d, allow_nd, ensure_min_samples, ensure_min_features, warn_on_dtype, estimator) 
    371          force_all_finite) 
    372  else: 
--> 373   array = np.array(array, dtype=dtype, order=order, copy=copy) 
    374 
    375   if ensure_2d: 

ValueError: setting an array element with a sequence. 
+0

Что форма 'trainingImages.image .values'? Значения X, переданные функции 'fit', должны иметь форму (n_samples, n_features). Если у вас есть (n_samples, width, height), попробуйте 'X.reshape (-1, width * height)'. – dukebody

ответ

1

Это, вероятно, потому, что «trainingImages.image.values» не имеют одинаковое число элементов во всех его массивов. Проверьте подобный вопрос здесь в StackOverflow:

ValueError: setting an array element with a sequence. while using SVM in scikit-learn

+0

Благодарим вас за ответ. Я убедился, что все его элементы имеют одинаковый размер, и я все равно получаю ту же ошибку. –

+0

Не могли бы вы опубликовать распечатку trainingImages.image.values ​​и trainingTargets.values.tolist()? (а также формы) – armatita

1

Если вы уверены, что размеры являются правильными, ниже это кусок кода/рабочего процесса, который может помочь

import skimage.io as skio 
import matplotlib.pyplot as plt 
import numpy as np 
from sklearn import svm 
from sklearn.metrics import accuracy_score 
from sklearn.metrics import precision_score 
%matplotlib inline 

# Load the data 
trainingImages = skio.imread_collection('train/images/*.jpg',conserve_memory=True) 

# cast to numpy arrays 
trainingImages = np.asarray(trainingImages) 

# reshape img array to vector 
def reshape_image(img): 
    return np.reshape(img,len(img)*len(img[0])) 

img_reshape = np.zeros((len(trainingImages),len(trainingImages[0])*len(trainingImages[0][0]))) 

for i in range(0,len(trainingImages)): 
    img_reshape[i] = reshape_image(trainingImages[i]) 

# SVM 
clf = svm.SVC(gamma=0.01,C=10,kernel='poly') 
clf.fit(img_reshape,trainingTargets.values.tolist()) 
Смежные вопросы