Tonybb9089

3-3-bilstm-torch comment error

1

class BiLSTM(nn.Module): def __init__(self): super(BiLSTM, self).__init__() self.lstm = nn.LSTM(input_size=n_class, hidden_size=n_hidden, bidirectional=True) self.W = nn.Parameter(torch.randn([n_hidden * 2, n_class]).type(dtype)) self.b = nn.Parameter(torch.randn([n_class]).type(dtype)) def forward(self, X): input = X.transpose(0, 1) # input :...

data_n.reset_index(drop=True, inplace=True) print("Now column number is: {}".format(data_n.shape[0])) 这里应该是row number 而不是column number更合理