Closed levtelyatnikov closed 10 months ago
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
All modified and coverable lines are covered by tests :white_check_mark:
Comparison is base (
e160cb8
) 97.49% compared to head (62513dc
) 97.45%.
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
Overall: I have reviewed the hypergraph implementation and made necessary modifications to ensure consistent patterns. Additionally, I have updated tutorials and test files to align with these changes.
Specifically: Previously, all the models generated the 'desired' hidden representation, meaning that if a model was required to perform a graph classification task, the pooling operation was executed within the model. Hence, the model output was the pooled signal.
Now, the models output the last hidden representations. In hypergraph models, the output includes nodes and hyperedges.
We have introduced a separate, lightweight pooling/readout class that is responsible for providing the desired output for particular downstream tasks. This new pooling/readout class is currently introduced in the tutorials.