通用方法為fluid.layers.accuracy(logits, label),其中logits為預(yù)測(cè)值,一般通過model(img)得到,label為圖片標(biāo)簽。
label要求的數(shù)據(jù)格式為int64,如不是,可通過fluid.layers.cast(label, dtype='int64'進(jìn)行轉(zhuǎn)換。
logits的變式有如下三種:
1.直接用logits
fluid.layers.accuracy(logits, label)
2.使用softmax
fluid.layers.accuracy(fluid.layers.softmax(logits), label)
3.使用sigmoid
fluid.layers.accuracy(fluid.layers.sigmoid(logits), label)
PS:個(gè)人認(rèn)為三種方式?jīng)]啥區(qū)別,結(jié)果都一樣。