pytorch tree lstm package
Project description
Tree LSTM
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
Installation
pip install tree-lstm
after installed, you can use this via
import TreeLSTM, Tree, BatchedTree
Usage
- 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 test.py
Thanks for reporting issues / contributions / starts / watches!
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for tree_lstm-0.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6aca5ff82ac7fa2be6f721cf5d475a4a752ef324b91f89fd471996a2737dfa73 |
|
MD5 | d2f40707e0f504a0b7e32d2078856554 |
|
BLAKE2b-256 | 473e13931eb964af9342d18f33a93675f5699d7f464e4adda4e9bdad8378d6b4 |