Code for the tag clustering task