Deleting func for predicted labels too