Renu Khandelwal
1 min readFeb 26, 2019

--

Hi Shee,

The problem you are asking is for multiclass classification of images.

For binary classification we use sigmoid activation function

classifier.add(Dense(units=1, activation='sigmoid'))

For multiclass classification, as in your example where we need to classify dataset of 5 animals, we will use Softmax activation function.

The above code would now change to

classifier.add(Dense(units=5, activation='softmax'))

Hope this helps

--

--

Renu Khandelwal
Renu Khandelwal

Written by Renu Khandelwal

A Technology Enthusiast who constantly seeks out new challenges by exploring cutting-edge technologies to make the world a better place!

No responses yet