在tf.contrib.learn.DNNClassifier中使用自定義閾值？

我正在處理二元分類問題，我在TensorFlow中使用了tf.contrib.learn.DNNClassifier類。當僅爲2個類調用這個估計器時，它使用0.5的閾值作爲2個類之間的截止值。我想知道是否有辦法使用自定義閾值，因爲這可能會提高模型的準確性。在tf.contrib.learn.DNNClassifier中使用自定義閾值？

我搜遍了網絡，顯然沒有辦法做到這一點。

任何幫助將不勝感激，謝謝。

來源

2017-07-01 the_midget_17

tf.contrib.learn.DNNClassifier類有一個名爲predict_proba的方法，該方法返回屬於給定輸入的每個類的概率。然後，您可以使用類似tf.round(prob+thres)的二進制閾值與自定義參數thres。

來源

2017-07-01 05:46:59

哇我在任何文檔中都沒有看到這個，我現在就試試看。謝謝！ –

實際上，現在我回到手冊頁，我記得看到它，但我沒有想過如此使用它 –

如果以上解決了您的問題，你可以考慮通過檢查標記接受答案。 –

在tf.contrib.learn.DNNClassifier中使用自定義閾值？

回答

相關問題