Question

嘿，我已经在张量流中构建了一个神经网络，我想用来对声音进行分类，我希望看到预测的结果（标签和标签的分数）

我正在用这一行获得分数

values，indices = tf.nn.top_k（x，10）//其中x是输入

并且我能够得到分数而不是相关的标签是否有任何我可以附加到这个或标签我必须定义以将分数映射到标签？

Answer 1

您可以通过检索sess.run的输出来获取预测和标签。

x = tf.placeholder(...) # x inputs
y_true = tf.placeholder(...) # labels

logits = ... # neural network model

predictions = tf.nn.softmax(logits) # prediction tensor

init_op = tf.initialize_all_variables()

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

    # get run outputs
    pred, labels = sess.run([predictions, y_true],
                            feed_dict={x: x_inputs, y_true: y_inputs) 

# do something with the outputs
print labels
pd.Dataframe(data=pred)

tensorfllow预测和标签

1 个答案: