在 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

如果喜欢请点赞！