I am using UNet for a segmentation task. The problem I have is that I\'m trying to use recall metric and I have 4 labels and I don\'t know how to make it work for more that 2 la