在tensorflow -Python中加载csv文件

Question

我正在使用tensorflow作为文本分类器代码。我正在学习的教程使用的是Kaggle的imdb数据集，并且加载它的代码是：

import tensorflow as tf
from tensorflow import keras
data = keras.datasets.imdb
(train_x,train_y),(test_x,test_y) = data.load_data()

现在，我有自己的CSV文件，我想将其用作数据集，如何加载该文件？等价的东西：

data='mydata.csv'
(train_x,train_y),(test_x,test_y) = data.load_data()

数据集包含2列（审阅，响应）。 “ mydata.csv”的格式也相同，均包含2列和纯文本数据。

Answer 1

尝试一下

import pandas as pd
df=pd.read_csv("mydata.csv",delim_whitespace=True,header=None)
col_name=['review','response']
df.columns=col_name

如果喜欢，请投票！