Imported from https://github.com/prakashpandey9/Text-Classification-Pytorch