python - NumPy 相当于 Keras 函数 utils.to_categorical

Question

我有一个使用 Keras 进行机器学习的 Python 脚本。我正在构建 X 和 Y，它们分别是特征和标签。

标签是这样构建的:

def main=():

   depth = 10
   nclass = 101
   skip = True
   output = "True"
   videos = 'sensor'
   img_rows, img_cols, frames = 8, 8, depth
   channel = 1 
   fname_npz = 'dataset_{}_{}_{}.npz'.format(
    nclass, depth, skip)

   vid3d = videoto3d.Videoto3D(img_rows, img_cols, frames)
   nb_classes = nclass

   x, y = loaddata(videos, vid3d, nclass,
                    output, skip)

   X = x.reshape((x.shape[0], img_rows, img_cols, frames, channel))
   Y = np_utils.to_categorical(y, nb_classes) # This needs to be changed

Keras中使用的函数“to_categorical”解释如下:

to_categorical

keras.utils.to_categorical(y, num_classes=None)

Converts a class vector (integers) to binary class matrix.

现在我正在使用 NumPy。能否让我知道如何构建同一行代码才能工作？换句话说，我正在寻找与 NumPy 中的“to_categorical”函数等效的函数。

Answer 1

这是一种简单的方法:

np.eye(nb_classes)[y]