用很多组50*27的时间片数据作为训练集输入网络，得到4分类结果，神经网络的参数应该怎么设置？

Question

我有一组按时间分布的放电数据，每个数据中有27组信号作为特征属性，数据有4种分类结果，现在我把这段时间数据每50个时间步组成一组作为训练数据来训练模型，请问如果使用一维卷积的话，nn.Conv1d的参数应该怎么写呀？如果使用LSTM，nn.LSTM的参数应该怎么写？
即，用很多组50*27的时间片数据作为训练集输入网络，得到4分类结果，神经网络的参数应该怎么设置？
能不能顺便解释一下pytorch里输入通道、输出通道、输入特征、时间片长度、batch_size这些参数的意思？

Answer 1

引用新必应 对于这个问题，我可以回答。如果使用一维卷积，nn.Conv1d的参数应该设置为nn.Conv1d(27, 64, kernel_size=3, stride=1, padding=1)，其中27是输入特征的数量，64是输出通道的数量，kernel_size是卷积核的大小，stride是步长，padding是填充大小。如果使用LSTM，nn.LSTM的参数应该设置为nn.LSTM(27, 64, num_layers=2, batch_first=True)，其中27是输入特征的数量，64是隐藏层的大小，num_layers是LSTM的层数，batch_first表示输入数据的维度顺序为(batch_size, seq_len, input_size)。

在PyTorch中，输入通道指的是输入数据的特征数量，输出通道指的是卷积层或线性层的输出特征数量，输入特征指的是每个数据点的特征数量，时间片长度指的是每个数据点的时间步数，batch_size指的是每个batch中的数据数量。