下图显示了在 CIFAR 小图像数据集上训练简单的深度 CNN(卷积神经网络)的部分代码。我已经导入了,导入keras.utils(以红色突出显示)
但是,我仍然收到以下错误:
您可以to_categorical在 keras 中导入,如下所示。
from keras.utils.np_utils import to_categorical
Run Code Online (Sandbox Code Playgroud)
它可以如下所示使用。
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)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
6632 次 |
| 最近记录: |