[問題] TensorFlow如何設定複雜的loss function?

作者: dogs1231992 (河豚)   2019-02-01 05:02:10
各位大家好,小弟最近想用TensorFlow來完成NN的部分
目前數據都準備好了,TensorFlow的部分則是上網查了一些資料,層及進出都設定好了
只是我遇到一個關於loss function的問題
一般訓練都是N個數據進去,M個數據出來,用這M個數據與參考值求誤差
但我的狀況是:
我有一個三維的資料: N x M x P
裡面有 N 個images,每個images有 M 個原子,每個原子有 P 個參數
希望可以利用同一個 (或者依照原子不同而分類成多個) NN
在我輸入P個數據後,可以得到答案Ep
同一個image裡面所有原子M的Ep加總後會得到一個值 En
全部資料會有N個En,這些En與參考值求誤差後即為我的loss
目前卡在不知道該如何將M次計算後的Ep全部加總在一起
作者: djshen (djshen)   2019-02-01 08:15:00
數學式寫出來阿
作者: f496328mm (為什麼會流淚)   2019-02-01 14:09:00
你loss function要設計的有彈性,不要寫死
作者: yoyololicon (蘿莉大好)   2019-02-01 16:29:00
你有設定batch size 嗎?還是都一個個餵?老實說你的loss function 看起來蠻蠻基本的,不懂為何會卡住我會建議你不要用一個個原子去想 而是用image的角度因為你的目標En單位是每個image就直接用(N, M, P)大小的矩陣直接去train不用拆開對應的label為長度(N,)的矩陣 這邊N就是batch size

Links booklink

Contact Us: admin [ a t ] ucptt.com