Получение ошибки в одной функции горячего кодирования, реализованной с помощью np.eye () - PullRequest
Новая
       7

Получение ошибки в одной функции горячего кодирования, реализованной с помощью np.eye ()

0 голосов
/ 05 декабря 2018

Мои каталоги тестов и поездов следующие:

  • поезд
  • - class_0
  • - class_1
  • test
  • - class_0
  • - class_1

train --class_0 --class_1 - test --class_0 --class_1

все классы содержат 2 изображения натеперь для тестирования, а также обучения.

Я использую следующий код, чтобы получить одну горячую кодировку для выходных меток 

def convert_to_one_hot(Y, C):
  print ("hehe ",Y.reshape(-1), np.eye(C))
  Y = np.eye(C)[Y.reshape(-1)].T
  return Y

В основной программе я передаю C = 2 и Y = [1 1 2 2] ипри этом я получаю следующую ошибку: 

---> 55         Y = np.eye(C)[Y.reshape(-1)].T
IndexError: index 2 is out of bounds for axis 0 with size 2

Я не могу исправить эту ошибку.Пожалуйста, помогите!

...