如何处理pytorch中的多特征预测?

时间:2019-07-14 11:21:00

标签: neural-network lstm recurrent-neural-network loss-function

我正在尝试开发一个顺序模型(使用LSTM s),该模型必须在每个时间步预测多个功能。我本来想将NLLLoss用作损失函数,但是我非常确定如何处理这种多特征的情况。

模型的输入对应于连接在一起的不同的一键编码特征。 在文档中,据说NLLLoss可以接受(N,C,d1,d2,...,dK)形式的输入,在此我假设必须为每个功能创建不同的d维才能预测?我试过了,但是没用。

我使用的是错误的损失函数,还是有针对特定情况使用NLLLoss的正确方法? 我在Google上搜索了很多,但仍然迷路了……

0 个答案:

没有答案