Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implement TokenGT in PyG #9856

Open
wants to merge 36 commits into
base: master
Choose a base branch
from
Open

Implement TokenGT in PyG #9856

wants to merge 36 commits into from

Conversation

KunalSinha1125
Copy link

  • Implementing TokenGT as described here (https://arxiv.org/abs/2207.02505)
  • Adding new files tokenGT.py in torch_geometric/nn/models/tokenGT and test_tokenGT.py in torch_geometric/nn/models/tokenGT

KunalSinha1125 and others added 23 commits November 5, 2024 18:22
@KunalSinha1125
Add token_gt_encoder.py first pass
9fd04c3
@salizade22
Create README.md
79882b5
@salizade22
Add files via upload
1e81d05
@salizade22
Create README.md
5bb74d6
@salizade22
Add files via upload
e0c895d
@salizade22
Delete torch_geometric/contrib/nn/models/tokenGT/tokenGT.py
f8215c2 
Reupload
@salizade22
Add files via upload
3f2507d
@paulwrg
fix dimension of P
dfd6b36
@paulwrg
switch to single input dim
646027e
@paulwrg
fixed P dimension (again but for d_p < num_nodes)
e8bdc3e
@paulwrg
convert indices to int
1d33bb9
@paulwrg
Fix node identifiers: generate them indepently in each graph, match i…
3d8a27d 
…mplementation with notes in appendix
@KunalSinha1125
Clean up testing code
a2f690a
@KunalSinha1125
Test multi-graph batches
74fc946
@salizade22
Update tokenGT.py
c0bd0f5
@salizade22
Update tokenizer.py
80a3b64
@salizade22
Update test_tokenGT.py
3f15f8f
@paulwrg
fixed tokenGT args
c5beffa
@salizade22
Update tokenizer.py
3f28072
@salizade22
Update tokenizer.py
cae08fb
@paulwrg
Moved files to their correct places
768ead5
@paulwrg
Formatting test file
a1fa70e
@paulwrg
fix package init for doc
723b52f
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@wsad1 wsad1 Awaiting requested review from wsad1 wsad1 is a code owner

@rusty1s rusty1s Awaiting requested review from rusty1s rusty1s is a code owner

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@KunalSinha1125 @salizade22 @paulwrg