How to predict with model when I have used tf.dataset to split dataframe?

Question

I\'m new to tensorflow & Keras and I\'m trying a tutorial from tf. However, they are using tf.dataset to split dataframe to train, test and validation. I wanted to add f