Code for Word-Based Heirarchical Classification model