https://github.com/ttpro1995/TreeLSTMSentiment/blob/c61f73e3efdf227e2cb521157f320ec727f2490a/trainer.py#L73 We need to do this only for binary classification. But for fine-grained it should be removed.