Skip to main content

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

  1. make Tree object (initialize with arbitrary tensor)
  2. make BatchedTree object using list of Tree object
  3. make TreeLSTM object (inherited from torch.nn.Module)
  4. 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

tree_lstm-0.0.7.tar.gz (3.0 kB view hashes)

Uploaded Source

Built Distribution

tree_lstm-0.0.7-py3-none-any.whl (4.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page