Function y2indicator

cnn_class2/fashion.py:18–23

(Y)

# helper
def y2indicator(Y):
  N = len(Y)
  K = len(set(Y))
  I = np.zeros((N, K))
  I[np.arange(N), Y] = 1
  return I


# get the data