Open naeemulhassan opened 9 years ago
@naeemulhassan Mind posting your code so I can try to reproduce this error?
I turned the gender two class problem into a multiclass problem, by replacing sigmoid
with softmax
, setting the size
of the final dense layer equal to 2
, and bce
(binary cross entropy) with cce
(categorical cross entropy).
It's important that the multi-class problem is handled through a softmax, because the softmax outputs a normalized distribution of probability values for the classes. I hope this was the simple change you needed! Let me know and we can try to hash it out together.
Check out the gender_multiclass.py
file in my PR: https://github.com/IndicoDataSolutions/Passage/pull/39
Here is the thing.
I have 3 class labels in my data. The predict function returns an array of shape (N, 3) where N is the size of the train set. Here is an example,
How to interpret the prediction. I have 3 class labels but the array is giving close to 1 in all the classes. The value don't even sum up to 1. I must be doing something wrong. Any suggestion is highly appreciated.
Thanks!