首页 > 解决方案 > Keras.utils 无法导入

问题描述

下图显示了在 CIFAR 小图像数据集上训练简单深度 CNN(卷积神经网络)的部分代码。我已经导入,导入 keras.utils(以红色突出显示)

在此处输入图像描述

但是,我仍然收到以下错误:

在此处输入图像描述

标签: python-3.xkeras

解决方案


您可以在 keras 中导入to_categorical,如下所示。

from keras.utils.np_utils import to_categorical

它可以如下图所示使用。

Y = [1, 2, 1, 2, 3, 4, 1]

Y = to_categorical(Y)

print(Y)

# output
array([[0., 1., 0., 0., 0.],
       [0., 0., 1., 0., 0.],
       [0., 1., 0., 0., 0.],
       [0., 0., 1., 0., 0.],
       [0., 0., 0., 1., 0.],
       [0., 0., 0., 0., 1.],
       [0., 1., 0., 0., 0.]], dtype=float32)

推荐阅读