2017-08-07 181 views
0

我看到新的DNN和估計班1.3(tf.estimator.DNNClassifier),但我沒有看到任何方式獲取/設置predict_proba ..TensorFlow Estimator 1.3無法獲得predict_proba?

所以截至目前的預測是在超過設置爲true .5%,假如我認爲是二進制情​​況下的錯誤,但是有很多用例預測真實值的概率是非常有用的,特別是對於非常不平衡的數據?

任何意見或建議?或指導?

BTW它可以在這裏找到: 類tf.contrib.learn.DNNClassifier

+0

關於得到'從'graph' inputs'和「predictions''tensor'和手動運行預測什麼。 –

+0

鄧諾,這將如何工作? (我正在努力與更高的api!) – dartdog

+0

使用檢查點'。* meta'文件,您可以得到保護程序並從同一個檢查點恢復權重。然後使用'graph.get_tensor_by_name'來獲得輸入和預測張量。 –

回答