Multi-class classification with Multi-hot encoded vector

Have a similar issue, did you figure it out?