code: def __init__(self, vocab_size=1308843, embed_dim=32, num_class=4):
      super(TextSentiment, self).__init__()
      self.embedding = nn.EmbeddingBag(
          vocab_size, embed_dim, sparse=True
      )
      self.fc = nn.Linear(embed_dim, num_class)
      self.init_weights()
  
imports: import torch.nn
label_desc: Use image node_id 1 to create a new TextSentiment object from inherited base classes: nn with example: obj = TextSentiment(vocab_size, embed_dim, num_class)