首先,您需要使用 Python 的 csv 模块来读取 CSV 文件。接下来,您可以使用 TensorFlow 的 Dataset API 建立自己的数据集。
例如,假设您的 CSV 文件中有两列,分别是数值型特征和标签,您可以使用以下代码来建立数据集:
import tensorflow as tf
import csv
# 读取 CSV 文件
data = []
with open('data.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
for row in reader: