Tensorflow numpy到tens

Question

我读过一个使用numpygenfromtxt的csv文件

csv_file = np.genfromtxt(args.dataset, delimiter=',',skip_header=1,usecols=(0,1,2,3,4,5), dtype=None)

问题：如何使用string_input_producer对文件进行排队和批处理。在

Answer 1

你可以像你一样从CSV读取Numpy数组，然后手动将其分割成若干批。然而，TF有一个内置的功能，可以从多个CSV文件中读取数据，并将行组合成随机或连续的批处理。您可以读取不同数据类型的单元格，并根据需要将它们转换为相关的数据类型。在

在本问题中讨论了执行此操作的工作代码： Converting TensorFlow tutorial to work with my own data

简而言之，您需要的关键函数是tf.TextLineReader，tf.train.string_input_producer，和{}，或者{}，这取决于您的需要。在

我知道这种方法的唯一限制是CSV文件中的行的长度应该相等。在