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

Refactor node attributes #64

Draft
wants to merge 5 commits into
base: develop
Choose a base branch
from

Conversation

JPXKQX
Copy link
Member

@JPXKQX JPXKQX commented Oct 18, 2024

This PR introduces the NamedNodesAttributes(nn.Module) class, designed to enhance the management of node attributes within graph neural networks. Key features include:

  • Node attribute tensor: The class provides a tensor that combines coordinates with trainable parameters for each set of nodes in the graph.
  • Grid size: It retains the grid_size attribute, indicating the total number of nodes.
  • Channel dimension: It tracks in_channels_src/dst, which represents the dimensionality of the node attribute tensor.

The main improvement is the ability to handle an arbitrary number of nodes. This development will facilitate extending AnemoiModelEncProcDec to other use cases, such as hierarchical models or multi-encoder/decoder.

@codecov-commenter
Copy link

codecov-commenter commented Oct 18, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.84%. Comparing base (0122962) to head (f10810c).

Additional details and impacted files
@@           Coverage Diff            @@
##           develop      #64   +/-   ##
========================================
  Coverage    99.84%   99.84%           
========================================
  Files           23       23           
  Lines         1301     1301           
========================================
  Hits          1299     1299           
  Misses           2        2           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@JPXKQX JPXKQX self-assigned this Oct 21, 2024
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.

2 participants