Загрузка CSV-файлов в tenorflow - Python

Question

Я использую тензор потока для кода классификатора текста. Учебник, который я изучаю, использует набор данных imdb из Kaggle, и код для его загрузки:

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

Делайте upvote, если хотите!