神经网络
文章平均质量分 62
神经网络知识
emPoint
每天进步一点点
展开
-
pytorch RuntimeError: invalid argument 1: .
当使用torch.view()时出现 RuntimeError: invalid argument 1: input is not contiguous at /pytorch/torch/lib/TH/generic/THTensor.c:231可以使用 tensor.contiguous() 解决mask[idx, :].view(bat_size, 1) 变成 mask[idx,...原创 2019-01-16 09:59:29 · 2694 阅读 · 1 评论 -
Pytorch 计算误判率,计算准确率,计算召回率
无论是官方文档还是各位大神的论文或搭建的网络很多都是计算准确率,很少有计算误判率,下面就说说怎么计算准确率以及误判率1.计算正确率获取每批次的预判正确个数train_correct = (pred == batch_y.squeeze(1)).sum()该语句的意思是 预测的标签与实际标签相等的总数获取训练集总的预判正确个数train_acc += train_correct.data[0] ...原创 2018-06-09 12:30:22 · 30474 阅读 · 13 评论 -
神经网络调参,提升准确率
------如果只是调cnn网络的参数,可以看这篇文章 http://cs231n.github.io/neural-networks-2/#init------当初步搭建好网络,模型能出初步结果后,若想优化,可以考虑调整:1.数据集扩增可以考虑增加数据集容量,有时候简单的增加容量,准确率提升明显2.增大数据集差异性,随机性在制作数据集时可以考虑增加数据的差异性3.tensor transform...原创 2018-06-05 16:09:26 · 24479 阅读 · 0 评论