首页 PaddleCV 帖子详情
零基础实战深度学习,关于用LeNet模型在眼疾识别的案例的问题
收藏
快速回复
PaddleCV 问答学习资料图像分类 757 3
零基础实战深度学习,关于用LeNet模型在眼疾识别的案例的问题
收藏
快速回复
PaddleCV 问答学习资料图像分类 757 3
# 运行模型前向计算,得到预测值
logits = model(img)
# 二分类,sigmoid计算后的结果以0.5为阈值分两个类别
# 计算sigmoid后的预测概率,进行loss计算
pred = F.sigmoid(logits)
loss = F.binary_cross_entropy_with_logits(logits, label)
# 计算预测概率小于0.5的类别
pred2 = pred * (-1.0) + 1.0
# 得到两个类别的预测概率,并沿第一个维度级联
pred = paddle.concat([pred2, pred], axis=1)
acc = paddle.metric.accuracy(pred, paddle.cast(label, dtype='int64'))

本人小白一个,跟着官网的教程一步一步来的,但是这里依然不是很懂,为什么要使用sigmod函数,为什么又要进行级联,希望各位大佬能帮忙解答一下

0
收藏
回复
全部评论(3)
时间顺序
宇宙物语
#2 回复于2022-12

预测概率是0到1之间,要把数值放缩到概率区间上。sigmoid函数有这个功能肯定用他啊

0
回复
李长安
#3 回复于2023-01

厉害了

0
回复
wl_zzz
#4 回复于2023-02

sigmod函数是用作激活函数的,不过现在不怎么用了,都用relu了

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户