U-NET: Validation per class