如何使用tensorflow 2中的make_csv_数据集将多列作为标签读取？

我尝试使用以下代码（在Tensorflow教程中找到）从CSV文件读取数据：

def get_dataset(file_path, **kwargs):
  dataset = tf.data.experimental.make_csv_dataset(
      file_path,
      batch_size=5, # Artificially small to make examples easier to show.
      label_name=LABEL_COLUMN,
      na_value="?",
      num_epochs=1,
      ignore_errors=True, 
      **kwargs)
  return dataset

当有一列作为标签列时，它可以正常工作。但是，在我的CSV文件中，我有多个列作为标签（我有1008个功能和2个列作为标签）。我想知道如何使用make_csv_数据集读取数据

---

谢谢大家!

Tensorflow的

make_csv_dataset（）

目前还不支持多输出模型的数据（毕竟是实验性的）。您也可以使用pandas dataframe读入数据，然后使用

tf.data.Dataset.from\u tensor\u slices（）

方法获取数据集。我建议在创建pandas数据帧时对csv数据进行内存映射，因为这样会更快