如何找到非卷积模型tensorflow python的概率

Question

我尝试了以下方法来找到我创建的模型的probability：

X = tf.placeholder(tf.float32, [None, n_steps, n_inputs])
y = tf.placeholder(tf.float32, [None,n_outputs])
prediction=tf.argmax(y,1)

然后在尝试时代评估时：

print ("Probability Training: ", prediction.eval(feed_dict={X: x_train,y:y_train}, session=sess))})

我在我的 jupyter 笔记本中得到以下输出：

Probability Training:  [0 3 1 ... 2 2 2]

这不是概率，而是 输出中的最大值我猜的。

请帮我找出模型的概率。请注意，我的模型使用的不是convolution，而是一个简单的LSTM。

Answer 1

您的prediction 变量是一个argmax 评估，它将为您提供最高分的索引。使用 softmax 应该会产生所需的输出，因为 softmax 会将您的分数转换为可以解释为概率的东西：

X = tf.placeholder(tf.float32, [None, n_steps, n_inputs])
y = tf.placeholder(tf.float32, [None,n_outputs])
probabilities=tf.nn.softmax(y)

您仍然可以在稍后阶段在概率之上抛出一个 argmax 以提取最可能的类