I’ve just implemented a basic Pytorch model and now I want to try a model that predicts

Multiclass output ie instead of just one output y_hat , it could be varying number of outputs. for eg : for a picture about the weather, it can predict [‘humid’,‘cold’,‘monsoon’] and for another just [‘hot’]

Multilabel output y_hat_1 and y_hat_2

How should the final layer look like, and what loss functions should I use? I’m still working through the course and the book, so apologies if this is already in there!