pytorch[6]

逻辑回归 ——做分类任务

pytorch[6]_第1张图片

torchvison 下载数据集                         (下载地址,下载测试集还是训练集,是否从网上下载)

 pytorch[6]_第2张图片

讲输出改为概率

pytorch[6]_第3张图片

pytorch[6]_第4张图片

loss变为如下  

pytorch[6]_第5张图片

pytorch[6]_第6张图片

 分类loss表示

pytorch[6]_第7张图片

二者比较,右侧多个F.sigmoid

pytorch[6]_第8张图片

pytorch[6]_第9张图片

你可能感兴趣的:(pytorch,人工智能,python)