Imported from https://github.com/radoslawkrolikowski/sentiment-analysis-pytorch