title: How to get the labels from tensorflow dataset
body: <pre><code>ds_test = tf.data.experimental.make_csv_dataset(
      file_pattern = &quot;./dfj_test/part-*.csv.gz&quot;,
      batch_size=batch_size, num_epochs=1, 
      #column_names=use_cols, 
      label_name='label_id',
      #select_columns= select_cols,
      num_parallel_reads=30, compression_type='GZIP',
      shuffle_buffer_size=12800)
  </code></pre>
  <p>This is my tesetset during training. After completing the model, I want to zip the columns of predictions and labels for the <code>df_test</code> .</p>
  <pre><code>preds = model.predict(df_test)
  </code></pre>
  <p>Getting the predictions is quite simple, and it is of numpy array format. However, I don't know how to get the corresponding labels from the df_test.
  I want to zip(preds, labels) for further analysis.
  Any hint? Thanks.</p>
  <p>(tf version 2.3.1)</p>