不同问题使用的损失函数和输出设置
回归问题:
损失函数:均方误差mse
输出层激活方式:无
二分类问题
回答是和否的问题
损失函数:BCEloss
输出层激活方式:sigmoid
输出张量长度为1
损失函数:CrossEntropyLoss
输出层激活方式:无
输出张量长度为2
多分类问题
多个分类的问题,输出与分类个数相同的长度的张量
损失函数:CrossEntropyLoss
输出层激活方式:无
损失函数 :nn.NLLLoss 要求labels必须是独热编码方式
输出层激活方式: torch.log_softmax