我可以一次加载张量流模型并多次使用吗

时间:2020-02-20 01:12:27

标签: python tensorflow

我正在使用tensorflow LSTM模型来预测股价。训练模型后,我通过SavedModelBuilder将其保存为:

x

然后,我加载模型进行回测,每小时进行一次预测,就加载一次模型。回测时间为12个月,因此我需要将模型加载8640次(= 12 * 30 * 24)。太费时间了。有没有办法一次加载模型并多次使用?

我的模型加载代码:

builder = tf.compat.v1.saved_model.builder.SavedModelBuilder(_MODEL_DIR)
builder.add_meta_graph_and_variables(session, [tf.saved_model.TRAINING])
builder.save()

我尝试过的事情:

  1. 我试图通过tf.Saver保存模型并加载它,它并不比SavedModelBuilder快。
  2. 我尝试加载一次模型,并多次使用。它的速度要快得多,但是除了第一个预测之外,我得到了错误的预测,就像这样:
for index in range(s_i, e_i + 1):
    tf.reset_default_graph() 
    with tf.compat.v1.Session() as sess:
        tf.compat.v1.saved_model.loader.load(sess, [tf.compat.v1.saved_model.TRAINING], model_dir)  #  load SavedModelBuilder model, this is the most time-consuming sentence
        result = order_decide(htd, sess, date, start_time=date[index])  # htd: historical market data, sess: tensorflow session, date: date list, start_time: the specified predicition time

我也阅读了以下帖子,但找不到合适的答案: Load a Tensorflow graph once and use it multiple times?

请帮助我。

0 个答案:

没有答案