亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

多熱標簽編碼

多熱標簽編碼

慕婉清6462132 2021-11-16 09:42:54
我是 Tensorflow 的新手。我有一個圖像數據集,其中一張圖像有多個標簽。據我了解,我需要使用tf.losses.sigmoid_cross_entropy(). 我嘗試應用于tf.one_hot標簽,但是當我嘗試將它們傳遞給損失函數時,我得到錯誤,形狀不兼容。我怎樣才能解決這個問題?
查看完整描述

1 回答

?
HUWWW

TA貢獻1874條經驗 獲得超12個贊

你是對的tf.losses.sigmoid_cross_entropy。所有你需要做的就是 wrap tf.one_hotwithtf.reduce_max來減少這樣的維度。

tf.reduce_max(tf.one_hot(labels, num_classes, dtype=tf.int32), axis=0)

那應該返回 shape 的張量(num_classes,),正是您的損失函數所需要的。


查看完整回答
反對 回復 2021-11-16
  • 1 回答
  • 0 關注
  • 273 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號