├── .gitignore ├── LICENSE ├── README.md ├── example └── tree_lstm │ ├── LICENSE │ ├── README.md │ ├── dataset.py │ ├── fetch_and_preprocess.sh │ ├── fold │ ├── lib │ ├── CollapseUnaryTransformer.java │ ├── ConstituencyParse.java │ └── DependencyParse.java │ ├── main.py │ ├── model.py │ └── scripts │ ├── download.py │ └── preprocess-sick.py ├── fold ├── __init__.py ├── fold.py └── rnn │ ├── __init__.py │ ├── fold_lstm.py │ └── fold_tree_lstm.py ├── img └── animation.gif ├── requirements.txt └── tests ├── dataset.py ├── fold ├── l_sentences.nd ├── model.py ├── r_sentences.nd ├── test_fold.py └── test_fold_tree_lstm.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/README.md -------------------------------------------------------------------------------- /example/tree_lstm/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/LICENSE -------------------------------------------------------------------------------- /example/tree_lstm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/README.md -------------------------------------------------------------------------------- /example/tree_lstm/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/dataset.py -------------------------------------------------------------------------------- /example/tree_lstm/fetch_and_preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/fetch_and_preprocess.sh -------------------------------------------------------------------------------- /example/tree_lstm/fold: -------------------------------------------------------------------------------- 1 | ../../fold -------------------------------------------------------------------------------- /example/tree_lstm/lib/CollapseUnaryTransformer.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/lib/CollapseUnaryTransformer.java -------------------------------------------------------------------------------- /example/tree_lstm/lib/ConstituencyParse.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/lib/ConstituencyParse.java -------------------------------------------------------------------------------- /example/tree_lstm/lib/DependencyParse.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/lib/DependencyParse.java -------------------------------------------------------------------------------- /example/tree_lstm/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/main.py -------------------------------------------------------------------------------- /example/tree_lstm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/model.py -------------------------------------------------------------------------------- /example/tree_lstm/scripts/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/scripts/download.py -------------------------------------------------------------------------------- /example/tree_lstm/scripts/preprocess-sick.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/example/tree_lstm/scripts/preprocess-sick.py -------------------------------------------------------------------------------- /fold/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/fold/__init__.py -------------------------------------------------------------------------------- /fold/fold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/fold/fold.py -------------------------------------------------------------------------------- /fold/rnn/__init__.py: -------------------------------------------------------------------------------- 1 | from .fold_tree_lstm import ChildSumLSTMCell 2 | -------------------------------------------------------------------------------- /fold/rnn/fold_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/fold/rnn/fold_lstm.py -------------------------------------------------------------------------------- /fold/rnn/fold_tree_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/fold/rnn/fold_tree_lstm.py -------------------------------------------------------------------------------- /img/animation.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/img/animation.gif -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | mxnet 2 | -------------------------------------------------------------------------------- /tests/dataset.py: -------------------------------------------------------------------------------- 1 | ../example/tree_lstm/dataset.py -------------------------------------------------------------------------------- /tests/fold: -------------------------------------------------------------------------------- 1 | ../fold -------------------------------------------------------------------------------- /tests/l_sentences.nd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/tests/l_sentences.nd -------------------------------------------------------------------------------- /tests/model.py: -------------------------------------------------------------------------------- 1 | ../example/tree_lstm/model.py -------------------------------------------------------------------------------- /tests/r_sentences.nd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/tests/r_sentences.nd -------------------------------------------------------------------------------- /tests/test_fold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/tests/test_fold.py -------------------------------------------------------------------------------- /tests/test_fold_tree_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/szha/mxnet-jit-batch/HEAD/tests/test_fold_tree_lstm.py --------------------------------------------------------------------------------