#124 · Primary category: NLP Tools & Text Processing

PyTorchText

fasttext lstm nlp pytorch textcnn textrcnn textrnn

1st Place Solution for Zhihu Machine Learning Challenge . Implementation of various text-classification models.

Project last updated:07/16/18

GitHub Stars

1.1K

Forks

360

Contributors

1

License

MIT

Why we included this project

Text classification in PyTorch usually means choosing between a handful of well-known architectures, and this repo makes that choice easier by shipping several of them in one place. The code comes from the first-place solution in the 2017 Zhihu Machine Learning Challenge and includes CNN, LSTM, RCNN, an Inception-style CNN, and FastText, all built on a shared base class with a common training and evaluation pipeline. That lets you run the same dataset through each model and compare results directly, which is exactly the experiment a team does before settling on a production approach. Along the way it shows practical techniques like character- versus word-level embeddings, data augmentation, fine-tuning pretrained checkpoints, and ensembling. The project has not been updated since 2018, so treat it as a study resource and a source of adaptable implementations rather than a maintained dependency, though the MIT license makes it easy to reuse the code in your own work.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category