問題タブ [cross-entropy]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
tensorflow - tf.losses.sigmoid_cross_entropy でパラメータの重みを設定するには?
私は今tf.losses.sigmoid_cross_entropy
、不均衡なデータセットで使用しようとしています。ただし、パラメーターの重みについては少し混乱しています。ドキュメントのコメントは次のとおりです。
weights: ランクが 0 またはラベルと同じランクのオプションのテンソルで、ラベルにブロードキャスト可能でなければなりません (つまり、すべての次元は 1 か、対応する損失次元と同じでなければなりません)。
tf.losses.softmax_cross_entropy
パラメータの重みで、各サンプルの重みを持つランク 1 のテンソルになることがわかっています。の重みがtf.losses.sigmoid_cross_entropy
ラベルと同じランクでなければならないのはなぜですか?
誰も私に答えることができますか?例を挙げた方が良いでしょう。