如何在Colab中使用Dataset API在Tensorflow中加载本地csv文件

Question

我正尝试使用以下命令将cov的csv文件上传到tensorflow

tensorflow.data.Dataset.from_tensor_slices(io.StringIO(uploaded[ 'data.csv'].decode('utf-8')))

我收到此错误

ValueError: Attempt to convert a value (<_io.StringIO object at 0x7f4bb80e8e58>) with an unsupported type (<class '_io.StringIO'>) to a Tensor.

如何解决？

Answer 1

您可以通过以下方式上传文件：

from google.colab import files
import pandas as pd
import io

uploader = files.upload()
train_data = pd.read_csv(io.StringIO(uploader['data.csv'].decode('utf-8')))