2017-08-07 12 views
0

予測がオーバーでtrueに設定されているので、今のようTensorFlow Estimator 1.3 predict_probaを取得する方法はありませんか?

..私は(tf.estimator.DNNClassifier)1.3の新DNNと推定クラスを参照してくださいが、私は/ predict_probaを設定し取得する方法が表示されません.5%の場合はfalseで、バイナリの場合はfalseです。しかし、非常に不均衡なデータの場合、真を予測する確率が低い場合は非常に便利です。

アドバイスやリードはありますか?または指導?

ところでそれがここにあります:GitHubのでGoogleのパー クラスtf.contrib.learn.DNNClassifier

+0

(...、predict_keys = "確率")を予測。 –

+0

Dunno、どうしたらいいですか? (私は上位のapiに苦労しています!) – dartdog

+0

チェックポイント '。* meta'ファイルを使用すると、セーバーを取得し、同じチェックポイントからウェイトを復元することができます。 'graph.get_tensor_by_name'を使って入力テンソルと予測テンソルを取得します。 –

答えて

関連する問題