将多个CSV文件导入Tensorflow数据集时为空数据集

时间:2019-12-27 20:23:34

标签: python tensorflow tensorflow-datasets

我正在尝试使用以下示例将csv文件导入一个数据集中并进行交织。

https://datascience.stackexchange.com/questions/51249/training-keras-model-with-multiple-csv-files

但是,当我尝试在示例中使用文件时,我得到了一个空数据集,无法下载它们。以下是我正在使用的代码的输出。我的数据集文件名为X_train_1.csv,至10。

<DatasetV1Adapter shapes: (), types: tf.string>
<DatasetV1Adapter shapes: (), types: tf.string>

这是我的代码。

import tensorflow as tf 
import os
def train_input_fn(features,labels, batch_size):

  filenames = []
  for i in range(1,11,1):
      filenames.append(os.getcwd()+"/"+features+str(i)+".csv")

  print(filenames)
  filenames = tf.data.Dataset.list_files(filenames)  
  features_dataset = filenames.interleave(lambda x: tf.data.TextLineDataset(x), num_parallel_calls=2) 
  tf.data.TextLineDataset(x), num_parallel_calls=32)
  print(features_dataset)

train_input_fn("X_train_","y_train_",8)


0 个答案:

没有答案