2017-02-27 4 views
1

私はテンソルフローを使って広告のctrを予測しています。google wide&deep modelです。 これは、広告がクリック(1)かそうでない(0)かを予測するための分類問題です。tensorflow estimator.pyでの分類の予測結果の確率を取得するにはどうすればよいですか?

しかし、1または0の結果は、P(X = 1)の場合と同様の確率で計算しなければならない> 0.2、結果は1になり、そうでなければ0

私の質問を取得する方法であります確率。

私はestimator.py [wide_n_deepモデルまたはワイドモデルのみを使用します]、関数predict、およびestimator.pyまたはgraph_actions.pyで呼び出されたいくつかの関数で多くの情報を表示しました。

しかし、それは動作しません。(予測結果も0または1、いない0または1の確率である)

をAPIドキュメントでは、predict_probaという名前のメソッドは、確率を返します。しかし、pred_proba = m.predict_proba(x=None, input_fn=lambda: input_fn(df_test))として使用すると、無限の長さの配列が返されます。この方法の使い方は? wide_n_deep_tutorial.py

ありがとう:

私のコードは以下に基づいて修正されました。

+0

することによりを交換して、あなたはより詳細な情報を指定してくださいでしょうか? –

+0

予測結果も0または1であり、0または1の確率ではありません。 – Cow

+0

評価メソッドをチェックして、いくつかのメトリックに確率が含まれているかどうかを確認できます。 –

答えて

0

pred_proba = m.predict_proba(x=None, input_fn=lambda: input_fn(df_test))

それは動作しませんどのように pred_proba = m.predict_proba(input_fn=lambda: input_fn(df_test))

関連する問題