TensorFlow如何输出预测

Question

我有例如X_train（70,513）X_test（70,2）Y_train（30,513）Y_train（30,2）如何获得我训练输出（0,1）数组值的张量流模型测试值。

Answer 1

由于你没有提供你的模型，我会假设你已经按标准方式完成了它。

我假设在设计模型后，您的输出存储在张量（例如）prediction中。

要查看预测，您需要在所需的占位符中创建会话和Feed。你就是这样做的：

sess = tf.Session()
sess.run(prediction, feed_dict={features: [[0.4, 1.1]]})

由于我不知道您的占位符是如何定义的，因此我将features作为占位符并将其作为数据矩阵。

确保使用

初始化变量

sess = tf.Session()
sess.run(tf.global_variable_initializer())