Thanks! Would you mind giving me an example with code? I’ve never done it.
What loss do you recommend? I’m currently using CrossEntropy.
I have one concern regarding the use of a single output, I’m worried that if the model only sees the class I’m trying to predict, it will predict class 0,2,3,4 as class 1 later on.