pytorch tree lstm package
This repository contains a Pytorch Implementation of "Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks " (https://arxiv.org/abs/1503.00075).
This contains two type of tree-lstm (Child sum, N-ary). This was tested by Python 3.6, Pytorch 1.3.0., and this internally uses dgl 0.4.0
This repository referenced https://github.com/dmlc/dgl/blob/master/examples/pytorch/tree_lstm/tree_lstm.py
pip install tree-lstm
after installed, you can use this via
import TreeLSTM, Tree, BatchedTree
- make Tree object (initialize with arbitrary tensor)
- make BatchedTree object using list of Tree object
- make TreeLSTM object (inherited from torch.nn.Module)
- pass BatchedTree object into TreeLSTM object
For more detailed usage, please see
Thanks for reporting issues / contributions / starts / watches!
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for tree_lstm-0.0.8-py3-none-any.whl