Skip to content

Dynamic Node Labels - #386

Merged
Jacob-Chmura merged 14 commits into
mainfrom
dev/node_label
Jan 17, 2026
Merged

Dynamic Node Labels#386
Jacob-Chmura merged 14 commits into
mainfrom
dev/node_label

Conversation

@Jacob-Chmura

Copy link
Copy Markdown
Member

Close #352

@Jacob-Chmura Jacob-Chmura self-assigned this Jan 15, 2026
@tgm-team tgm-team deleted a comment from codecov Bot Jan 16, 2026
@tgm-team tgm-team deleted a comment from codecov Bot Jan 16, 2026
@Jacob-Chmura
Jacob-Chmura marked this pull request as ready for review January 16, 2026 20:19
@tgm-team tgm-team deleted a comment from codecov Bot Jan 16, 2026
@codecov

codecov Bot commented Jan 16, 2026

Copy link
Copy Markdown

Codecov Report

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

Files with missing lines Patch % Lines
tgm/data/split.py 94.11% 2 Missing ⚠️
tgm/data/dg_data.py 99.10% 1 Missing ⚠️

📢 Thoughts on this report? Let us know!

@ntgbaoo ntgbaoo left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

batch.dst here need to be updated to batch.edge_dst

Comment thread tgm/data/dg_data.py Outdated
Comment thread tgm/data/dg_data.py Outdated
node_x_mask (Tensor | None): Indices of node events within `time`.
node_x_nids (Tensor | None): Node IDs corresponding to node events [num_node_events].
node_x (Tensor | None): Node features over time [num_node_events, D_node_dynamic].
node_y_mask (Tensor | None): Indices of node labels within `time`.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
node_y_mask (Tensor | None): Indices of node labels within `time`.
node_y_mask (Tensor | None): Mask of node events within `time`.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is called Dynamic node features for nodes in DGBatch.

Comment thread tgm/data/dg_data.py Outdated
Comment thread tgm/data/dg_data.py Outdated
Comment thread tgm/data/dg_data.py
Comment thread tgm/core/batch.py
Comment thread docs/tutorials/dgraph_tutorial.md Outdated
Comment thread docs/tutorials/dgraph_tutorial.md Outdated

@ntgbaoo ntgbaoo left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The APIs now look very nice and clear to me. Thanks @Jacob-Chmura for the changes

@Jacob-Chmura
Jacob-Chmura merged commit 66a2125 into main Jan 17, 2026
7 checks passed
@Jacob-Chmura
Jacob-Chmura deleted the dev/node_label branch January 17, 2026 18:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Distinguish between Node Labels and Node Features

2 participants