###############################################
put the dataset word50 here.
"word2label.pt" is for mapping the word to its corresponding label.
###############################################