You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

25 lines
723 B

4 years ago
4 years ago
4 years ago
4 years ago
  1. # pruned-nlu
  2. ## Requirements
  3. - \>= Python 3.7
  4. - \>= PyTorch 1.4.0
  5. - [seqeval](https://github.com/chakki-works/seqeval)
  6. ## Setup
  7. Download GloVe embeddings:
  8. ```
  9. wget http://nlp.stanford.edu/data/glove.6B.zip
  10. unzip glove.6B.zip -d glove
  11. ```
  12. ## Files
  13. - `train.py`, `test.py`, `distill.py`, `timer.py`, `prune.py`: runnable scripts, check each file's argparse for options and details
  14. - `models.py`: intent detection, slot-filling, and multi-task (joint intent detection and slot filling) CNN models
  15. - `dataset.py`: dataset loading abstractions
  16. - `util.py`: common code
  17. - `models/`: pretrained models, 5 duplicates of each
  18. - `datasets/`: prepared ATIS and Snips datasets
  19. ## Contact
  20. [ojas@utexas.edu](mailto:ojas@utexas.edu)