Overview
PyText is a deep-learning based NLP modeling framework built on PyTorch. PyText addresses the often-conflicting requirements of enabling rapid experimentation and of serving models at scale. It achieves this by providing simple and extensible interfaces and abstractions for model components, and by using PyTorch’s capabilities of exporting models for inference via the optimized Caffe2 execution engine. We are using PyText in Facebook to iterate quickly on new modeling ideas and then seamlessly ship them at scale.
Core PyText features:
- Production ready models for various NLP/NLU tasks:
- Text classifiers
- Sequence taggers
- Joint intent-slot model
- Contextual intent-slot models
- Distributed-training support built on the new C10d backend in PyTorch 1.0
- Mixed precision training support through APEX (trains faster with less GPU memory on NVIDIA Tensor Cores)
- Extensible components that allows easy creation of new models and tasks
- Reference implementation and a pretrained model for the paper: Gupta et al. (2018): Semantic Parsing for Task Oriented Dialog using Hierarchical Representations
- Ensemble training support
Installing PyText
PyText requires Python 3.6.1 or above.
To get started on a Cloud VM, check out our guide.
Get the source code:
$ git clone https://github.com/facebookresearch/pytext
$ cd pytext
Create a virtualenv and install PyText:
$ python3 -m venv pytext_venv
$ source pytext_venv/bin/activate
(pytext_venv) $ pip install pytext-nlp
Detailed instructions and more installation options can be found in our Documentation. If you encounter issues with missing dependencies during installation, please refer to OS Dependencies.
Train your first text classifier
For this first example, we'll train a CNN-based text-classifier that classifies text utterances, using the examples in tests/data/train_data_tiny.tsv
. The data and configs files can be obtained either by cloning the repository or by downloading the files manually from GitHub.
(pytext_venv) $ pytext train < demo/configs/docnn.json
By default, the model is created in /tmp/model.pt
Now you can export your model as a caffe2 net:
(pytext_venv) $ pytext export < demo/configs/docnn.json
You can use the exported caffe2 model to predict the class of raw utterances like this:
(pytext_venv) $ pytext --config-file demo/configs/docnn.json predict <<< '{"text": "create an alarm for 1:30 pm"}'
More examples and tutorials can be found in Full Documentation.
Join the community
- Facebook group: https://www.facebook.com/groups/pytext/
License
PyText is BSD-licensed, as found in the LICENSE file.