Aug 3, 2022
Hei Ai,
Good point. As I mentioned at the end of the article, their code is kind of confusing. However, I am sure the sigmoid function is not applied to the final output when num_labels is set to 1, as I get results that are larger than 1. If the sigmoid function is applied, the range will be contained (0,1), right? Let me know what you think. I will also be happy to test the difference between 0 and 1 later and get back to you on this.