Update README.md

README.md

@@ -33,14 +33,15 @@ augmented_sst2_dataset = load_dataset("jmamou/augmented-glue-sst2")
 tokenizer = Tokenizer(num_words=10000)
 tokenizer.fit_on_texts(augmented_sst2_dataset['train']['sentence'])
 
-# Encode training data sentences into sequences
+# Encode test data sentences into sequences
 test_sequences = tokenizer.texts_to_sequences(sst2['test']['text'])
 
-# Pad the training sequences
+# Pad the test sequences
 test_padded = pad_sequences(test_sequences, padding=pad_type = 'post', truncating=trunc_type = 'post', maxlen=64)
 
 reloaded_model = from_pretrained_keras('moshew/distilbilstm-finetuned-sst-2-english')
 
+#Evaluate model on SST2 test data (GLUE)
 pred=reloaded_model.predict(test_padded)
 pred_bin = np.argmax(pred,1)
 accuracy_score(pred_bin, sst2['test']['label'])