flow_from_directory class_mode 参数。

问题描述 投票:0回答:1

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

tensorflow machine-learning keras deep-learning conv-neural-network
1个回答
1
投票

选项有 "分类"、"二进制"、"稀疏"、"输入 "或无。

你需要的是稀疏的,即它会给你1D整数标签。

© www.soinside.com 2019 - 2024. All rights reserved.