Imported from https://github.com/CognitiveModeling/2019-ModeInferencePaperCode