python - 如何应用训练好的神经网络？

Question

[Python 3.7，Tensorflow]我已经训练了一个神经网络。一切都很好，它会学习，但是一旦完成学习，它就会关闭并且进度会丢失。现在我想做的是输入新数据并亲眼看看网络表现如何。

我已经摆弄过

saver = tf.train.Saver()
saver.save(sess, 'model/model.ckpt')

但这总是会导致一英里长的错误报告以“未知错误:无法重命名'model/model.ckpt'”等结尾。

上下文中的代码如下所示:

def train_neural_network(x):
    training_data = generate_training_data() # i cut getting training data since its a bit out of context here, but its basically like mnist data

    prediction = neural_network_model(x) # normal, 3-layer feed forward NN
    cost = tf.reduce_mean( tf.square(prediction - y) )
    optimizer = tf.train.AdamOptimizer(0.01).minimize(cost)

    hm_epochs = 10
    saver = tf.train.Saver()

    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())

        for epoch in range(hm_epochs+1):
            epoch_loss = 0

            for i in range(10):
                epoch_x, epoch_y = training_data

                _, c = sess.run([optimizer, cost], feed_dict = {x: epoch_x, y: epoch_y})

        saver.save(sess, 'model/model.ckpt')

我尝试在 main 中调用这个经过训练的神经网络:

train_neural_network(x)

X, Y = generate_training_data()

prediction = neural_network_model(x)
saver = tf.train.Saver()

with tf.Session() as sess:

     saver.restore(sess, 'model/model.ckpt')
     result = sess.run(prediction, feed_dict={x: X})

print(Y, result)

到目前为止，所有内容都在一个文件中，但我也可以使用两个单独的文件。

这会导致一个错误，该错误表示通常的 python 错误由其路径组成并以
结尾“...在 _do_call 中 引发类型(e)(node_def，op，消息)”我认为，在出现 Tensorflow 特定错误之前:“未知错误:无法重命名‘model/model.ckpt’”和“由操作‘save_13/SaveV2’引起，定义于:”，然后有一条很长很长的路，大约有87行长，再次重复“未知错误”。

我想要的是带有神经网络预测输出的打印标签。 (代码中的打印行。)

不幸的是，到目前为止我还没有找到任何适用于各种互联网搜索的东西，但我觉得让它工作应该不会太难。预先感谢您。

Answer 1

如果您查看模型输出检查点的文件夹 (/model)，您应该会看到每个保存有 3 个单独的文件:model.ckpt-xxx.data、model.ckpt-xxx.index 和 model。 ckpt-xxx.meta，其中xxx是Tensorflow附加的检查点的ID。

当您想要恢复某个检查点时，您还必须添加 ID，因为通常在训练期间会创建同一网络的多个检查点，以便我们可以在以后必要时重新训练网络。

所以我会查看模型文件夹并仔细检查文件名，我的猜测是如果您只创建了一个检查点，saver.restore(sess, 'model/model.ckpt-0') 就可以解决问题。