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.

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

rewrite recurrent temporal layers for Flux v0.16 #560

Merged
merged 11 commits into from
Dec 17, 2024
Merged

Conversation

CarloLucibello
Copy link
Member

@CarloLucibello CarloLucibello commented Dec 16, 2024

Porting the temporal layers to the recurrence style of Flux v0.16.

Implements the GNNRecurrence layer, similar to the next Flux.Recurrence layer but also taking the graph as input.

Also, we now export both the cells and the layer's constructor (based on GNNRecurrence) .

  • GNNRecurrence for GNNGraph and TemporalSnapshotsGNNGraph
  • GConvGRUCell and GConvGRU
  • GConvLSTMCell and GConvLSTM
  • DCGRUCell and DCGRU
  • EvolveGCNOCell
  • EvolveGCNO for temporal graph
  • A3TGCN original code and paper so poorly written that they don't deserve inclusion
  • TGCNCell and TGCN (changed, fix correctness temporalgraph implementation #548)
  • update tests
  • update docs
  • update tutorials

@CarloLucibello CarloLucibello merged commit 27d13c8 into master Dec 17, 2024
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

correctness temporalgraph implementation
1 participant