【pre】
代码没啥新意,很多人都分享过了。主要想记录一下自己遇到的bug以及收获。
【bug】
1. RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x5 and 6x3)
nn.linear() 中,两参数(矩阵)的维度不一致。
FC公式:y=w*x+b,所以这里报错的原因是w和x维度不一致。
print了一下,发现x是5列,但源代码里作者认为是6列——“五个特征转化为了6维,因为get_dummies将性别这一个特征用两个维度来表示,即男性[1,0],女性[0,1]”。
可能是数据集的差异吧,我这里print后发现,性别直接用了0,1来表示,所以还是5维。
于是把线性层的输入改成了5。
问题解决,跑起来了。
【tips】
1.离散数据编码方式总结
根据某一feature的值之间有无大小关系,从而决定使用数字编码或是one-hot编码方式。(