从pickle文件的tensorflow中高效加载数据

时间:2019-04-11 15:34:55

标签: python tensorflow pickle

我应该读取大约10000个泡菜文件,并将其输入到tensorflow模型中进行训练。我通过一次加载文件来循环执行此过程,并进行一些预处理,然后训练使用此文件并重复。有没有更有效的方法来做到这一点。 该部分的代码类似于:

for filename in os.listdir("training_data"):
    x = load_data(filename)
    #some preprocessing
    feed_dict = construct_feed_dictionary(x)
    out = sess.run([model.opt_op, model.loss, model.accuracy],feed_dict=feed_dict) 

0 个答案:

没有答案