我正在使用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()
我尝试过的事情:
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?
请帮助我。