Skip to content

Add TGN v2#410

Open
shenyangHuang wants to merge 1 commit into
mainfrom
tgnv2
Open

Add TGN v2#410
shenyangHuang wants to merge 1 commit into
mainfrom
tgnv2

Conversation

@shenyangHuang
Copy link
Copy Markdown
Collaborator

@shenyangHuang shenyangHuang commented May 15, 2026

Summary / Description

adding TGN v2
Related Issues: #409

Type of Change

  • Bug fix
  • New feature
  • Breaking Change
  • Refactoring
  • Documentation update

Test Evidence

Describe how this PR has been tested.

  • Unit tests
  • Integration tests
  • Performance tests

Questions / Discussion Points

On node property prediction task it gets to 34% val NDCG on tgbn-trade with 10 epochs, still might have some issues.

30 epochs 35.7% NDCG

@shenyangHuang shenyangHuang marked this pull request as ready for review May 15, 2026 15:12
@shenyangHuang shenyangHuang self-assigned this May 15, 2026
@shenyangHuang shenyangHuang added the nn Library network layers label May 15, 2026
@shenyangHuang shenyangHuang added this to the Full Release milestone May 15, 2026
@codecov
Copy link
Copy Markdown

codecov Bot commented May 15, 2026

Codecov Report

❌ Patch coverage is 92.50000% with 3 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
tgm/nn/encoder/tgn.py 90.62% 3 Missing ⚠️

📢 Thoughts on this report? Let us know!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

nn Library network layers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant