CODEWITHSUNDEEP
tensorflowmachine-learningkerasdeep-learningconv-neural-network
Roshan Nayak
Roshan Nayak

Reputation: 53

flow_from_directory class_mode parameter

I was writing a CNN model to classify if the image had a forest, glacier, etc. Totally there were 6 classes. Hence the problem is of multi-class classification problems. I wanted to know what different values can the class_mode parameter of flow_from_directory have. Which value of them fits for this problem. My model output layer will have one node that outputs 0, 1, 2, 3, 4, or 5 denoting the class that the image belongs to.

Upvotes: 0

Views: 469

Answers (1)

Abhishek Verma
Abhishek Verma

Reputation: 1729

The options are "categorical", "binary", "sparse", "input", or None.

What you need is sparse i.e. it will give you 1D integer labels.

Upvotes: 2

Related Questions