Shypd.aiThe largest open index of AI toolsHedwig is an open-source collection of PyTorch deep learning models for document classification, developed by the Data Systems Group at the University of Waterloo. It provides implementations of various models like DocBERT and Reg-LSTM for researchers and practitioners.
Trending
Also listed in
About
Hedwig is an open-source repository offering PyTorch deep learning models specifically designed for document classification tasks. Developed by the Data Systems Group at the University of Waterloo, it includes implementations of several prominent models such as DocBERT, Reg-LSTM, XML-CNN, HAN, Char-CNN, and Kim CNN. Each model directory contains a detailed README.md for further information. The project is designed for Python 3.6 and PyTorch 0.4, with clear instructions for environment setup using Anaconda and installation of dependencies. It also provides options for downloading necessary datasets like Reuters, AAPD, and IMDB, along with word2vec embeddings, making it a comprehensive resource for document classification research and application.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
