diff --git a/beginner_source/text_sentiment_ngrams_tutorial.py b/beginner_source/text_sentiment_ngrams_tutorial.py index d842a058325..35b8f8ec943 100644 --- a/beginner_source/text_sentiment_ngrams_tutorial.py +++ b/beginner_source/text_sentiment_ngrams_tutorial.py @@ -110,8 +110,8 @@ def forward(self, text, offsets): VOCAB_SIZE = len(train_dataset.get_vocab()) EMBED_DIM = 32 -NUN_CLASS = len(train_dataset.get_labels()) -model = TextSentiment(VOCAB_SIZE, EMBED_DIM, NUN_CLASS).to(device) +NUM_CLASS = len(train_dataset.get_labels()) +model = TextSentiment(VOCAB_SIZE, EMBED_DIM, NUM_CLASS).to(device) ######################################################################