Best way to evaluate performance with custom metrics using tf.data.Dataset

Question

I trained a model and now want to evaluate its performance on a test set. The test set is loaded as tf.data.TFRecordDataset object (from multiple TFRecords with mul