model.fit結合dataset實現輸入數據的正樣本加權

最近寫代碼想實現給我的loss正樣本加權，因爲我的數據集中正樣本佔比只有15.79%，爲了保證tensorflow的IO性能，我採取使用dataset構建高效的數據輸入流水線。 正在我憂愁要不要自己寫weight_loss函數的時候開心地發現model.fit提供了參數class_weight，只需要輸入字典結構的class_weight就可自動實現對已有的loss進行分類別的加權。 （不要以爲w

>>阅读原文<<