Sentiment analysis

Guys, help solve the problem. I took the nn.transformer module. At the entrance is a sub-offer, at the output I need to get an analysis, positive or negative. I want to use softmax (0,1). How can I do it? nn.transformer for decoder is a required parameter (tgt - the sequence to the decoder (required)). What should I apply to this input?

2 Likes