2017-08-07 183 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'来获得输入和预测张量。 –

回答