通用方法为fluid.layers.accuracy(logits, label)
,其中logits为预测值,一般通过model(img)
得到,label为图片标签。
label要求的数据格式为int64,如不是,可通过fluid.layers.cast(label, dtype='int64'
进行转换。
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:个人认为三种方式没啥区别,结果都一样。
网友评论