Обучение классификатора SVM на паре вложений, генерирующих ошибку - PullRequest
Новая
       15

Обучение классификатора SVM на паре вложений, генерирующих ошибку

0 голосов
/ 11 октября 2018

Для каждого человека в моем наборе данных у меня есть помеченная коллекция пары изображений из их разных профилей. Для человека A данные выглядят следующим образом:

PersonA FacebookImage InstagramImage 0 (0 - метка, еслиоба изображения принадлежат лицу А)

Я сгенерировал вложения (в виде массива с нулевыми массами) для каждого изображения длиной 128 и объединил их в массив с массивными парами в виде пары.

Основная идея состоит в том, чтобы обучить классификатор таким образом, чтобы два вложения изображений, принадлежащих одному и тому же человеку, были похожи.

Данные обучения выглядят примерно так: 

[[[0 3 4..............   [0 1 .........
   .............          ............
   .................      .............
   .................128]  .........128]]...................]]]

иданные меток обучения выглядят так: 

[0,1,1,0,0,.........]

, где каждая пара является парой вложений одного и того же человека.Я пытаюсь обучить эти вложения на классификаторе SVM, используя код:

clf = svm.SVC(kernel='linear', C=1).fit(X_train, y_train) , но у меня появляется сообщение об ошибке: 

Found array with dim 3. Estimator expected <= 2.

Как я могу устранить эту ошибку?

Форма моего тренировочного набора (2000,2128), а форма каждого вложения - (128,)

...