pred <- multiLayerPerceptron(test_data)
init <- tf$global_variables_initializer()
with(tf$Session() %as% sess, {
sess$run(init)
model_saver$restore(sess, "log_files/model_MLP1")
test_pred_1 <- sess$run(pred, feed_dict= dict(x = test_data))
})
代码一切正常吗?我想通过这部分代码来获得test_data模型的预测。
答案 0 :(得分:0)
您的代码未显示初始化model_saver
的位置,但应在创建计算图形后创建。如果没有,它不知道要恢复/保存哪些变量。因此,在model_saver
之后创建pred <- multiLayerPerceptron(test_data)
。
请注意,如果您在训练期间犯了同样的错误,您的检查点将为空,您需要先重新训练您的模型。