dan
/
pruned-nlu
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
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.
2 Commits
1 Branch
97 MiB
 
 
Tree: a41d921ed5
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a41d921ed5'
${ noResults }
 ZIP  TAR.GZ
Ojas Ahuja a41d921ed5 Update README 4 years ago
datasets Initial commit 4 years ago
models Initial commit 4 years ago
.gitignore Initial commit 4 years ago
README.md Update README 4 years ago
dataset.py Initial commit 4 years ago
distill.py Initial commit 4 years ago
models.py Initial commit 4 years ago
prune.py Initial commit 4 years ago
requirements.txt Initial commit 4 years ago
test.py Initial commit 4 years ago
timer.py Initial commit 4 years ago
train.py Initial commit 4 years ago
util.py Initial commit 4 years ago

README.md

pruned-nlu

Requirements

  • >= Python 3.7
  • >= PyTorch 1.4.0
  • seqeval

Setup

Download GloVe embeddings:

wget http://nlp.stanford.edu/data/glove.6B.zip
unzip glove.6B.zip -d glove

Files

  • train.py, test.py, distill.py, timer.py, prune.py: runnable scripts, check each file's argparse for options and details
  • models.py: intent detection, slot-filling, and multi-task (joint intent detection and slot filling) CNN models
  • dataset.py: dataset loading abstractions
  • util.py: common code
  • models/: pretrained models, 5 duplicates of each
  • datasets/: prepared ATIS and Snips datasets

Contact

ojas@utexas.edu