How to specify labels / features when training a keras model from tf.data.Dataset?

Question

I followed the official tutorial on how to create/load TFRecords here.

So, now my dataset contains examples in the form of {"feat1":..., "feat2"