我有一个预训练的 cnn,想看看模型对一些简单的 imdb 预测的信心。例如,我有一句话“我喜欢这部电影”,我想把它扔进我的 cnn 并得到 {"pos": x,"neg": y} 值。
到目前为止,这是我所拥有的:
我已将 glove.840B.300d
加载到字典中,其中 {'word': GloVE embedding}
我有以下预先训练好的火炬模型:
cnn(
(embs): Embedding(43036,300)
(linear_input): Linear(in_features=300,out_features=100,bias=True)
(conv1): Conv1d(100,100,kernel_size=(3,),stride=(1,))
(fc_hidden): Linear(in_features=100,bias=True)
(fc_output): Linear(in_features=100,out_features=1,bias=True)
)
如何将我的句子插入模型以从中获得所需的预测?