tensorflow - flow_from_directory 类模式参数

Question

我正在编写一个 CNN 模型来分类图像是否有森林、冰川等。总共有 6 个类。因此问题是多类分类问题。我想知道 flow_from_directory 的 class_mode 参数可以有哪些不同的值。它们的哪个值适合这个问题。我的模型输出层将有一个节点输出 0、1、2、3、4 或 5，表示图像所属的类别。

标签： tensorflowmachine-learningkerasdeep-learningconv-neural-network

Answer 1

选项为“分类”、“二进制”、“稀疏”、“输入”或“无”。

你需要的是稀疏的，即它会给你一维整数标签。