In preparation for the NAACL2019 camera ready and future work, @achyudh is currently working on moving the document classification models into Hedwig.