TensorFlow如何輸出預測

Question

我有例如X_train（70,513）X_test（70,2）Y_train（30,513）Y_train（30,2）我怎樣才能得到我訓練輸出的張量流模型（0,1 ）測試值的數組值。

Answer 1

既然你還沒有提供你的模型，我會假設你已經做到了標準的方式。

我假設在設計你的模型後，你的輸出存儲在張量（比如說）prediction中。

要查看預測，您需要創建一個會話並以所需的佔位符形式提供。這是你如何做到這一點：

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

因爲我不知道你的佔位符是如何定義的，我做了features的佔位符和feeded其數據的矩陣。

請確保您已使用

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