Khuyen Le
Jan 20, 2022

Actually, if you want to apply this model for binary classification, you must change the unit number of the last layer. You can either replace 1000 by 2 and keep the softmax function, or use the sigmoid function with only one output unit. The sigmoid function will return the probability that an object belongs to the positive class, while the softmax function returns directly the predicted class.

I hope this answer is helpful for your.

Khuyen Le
Khuyen Le

Written by Khuyen Le

Postdoctoral Researcher at 3IA Côte d'Azur - Interdisciplinary Institute for Artificial Intelligence

Responses (1)