Skip to content

Releases: JetBrains-Research/embeddings-for-trees

Embeddings-for-trees | 1.2.0

19 Nov 14:23
Compare
Choose a tag to compare
  • Build vocabulary based on counts of tokens
  • Detailed wandb configuration
  • Return attention weights from forward
  • Upgrade dependencies to Torch, DGL, Lightning, etc
  • Fix bugs

Fix reseting metrics

25 Sep 11:51
4609ec3
Compare
Choose a tag to compare

Call metric reset on each epoch end

Improved uploading model checkpoints

22 Sep 08:48
Compare
Choose a tag to compare
1.0.3

Update requirements, new logic for uploading weights

Add all modules into the package

16 Sep 08:36
Compare
Choose a tag to compare
1.0.2

Add init files for correct package resolving

First release of embeddings-for-trees

16 Sep 08:10
f833caa
Compare
Choose a tag to compare

The first release of embeddings for trees package.
This release includes TreeLSTM PyTorch modules and lightning module to train and evaluate it.