from setuptools import setup

setup(name='hedwig',
      version='2.0.0',
      description='PyTorch deep learning models for document classification',
      packages=['char_cnn', 'han', 'kim_cnn', 'reg_lstm', 'xml_cnn', 'bert', 'hbert'],
      )
