How to suppoer multiclass?

[PeiyanFlying]

Hi friends,

Can I ask about how to support multi-class?

Here is my mask:

Should I change the num_classes into 4? And metrics type into 'multiclass' as the same time?

When I do this, there are some size error:

Could you please give me some suggestions!

Thanks so much!

[PeiyanFlying]

This is my corresponding input:

[PeiyanFlying]

Or should I keep the num_classes as 1? And change the metrics type into 'multiclass'?

Thanks a lot!