如何检查经过训练的神经网络的预测输出

时间:2020-08-29 16:15:44

标签: python tensorflow nan

我正在使用此代码

https://medium.com/@curiousily/tensorflow-for-hackers-part-ii-building-simple-neural-network-2d6779d2f91b

此代码没有任何问题。我的问题是:

如何预测经过训练的网络的输出?我已经尝试过使用命令

pred_NN = predictions.eval(feed_dict = {x: x_test,keep_prob: 1})

但它返回

       [nan, nan],
       ..........
       [nan, nan]], dtype=float32)`

1 个答案:

答案 0 :(得分:0)

loss function is returning nan tensorflow)有一些答案。降低学习率以防止转换为无穷大可能也有帮助。