类别数量,keras

时间:2017-07-06 07:10:34

标签: keras

我有一个keras模型,可以将图像分类为多个类别。如何打印模型将图像分类的类别数?

1 个答案:

答案 0 :(得分:2)

您可以随时使用Keras中的model.summary()检查模型的结构,它会按顺序为您提供模型中每一层的信息,如下图所示。enter image description here

所以在最后一层,也就是我的输出层,你可以看到输出形状是(None,10),它显示我的分类器包含10个类或10个类别。 (不要担心没有它只是意味着批量)。