-
Notifications
You must be signed in to change notification settings - Fork 96
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
Particlenet added and all km3net dependencies are deleted #740
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hey @giogiopg @IvanMM27 - thanks for this very clean pull request 🚀 !
There is only a few minor changes needed before we can merge this.
ParticleNeT.py
should be changed toparticlenet.py
orparticle_net.py
so that it is in line with the snake case convention. (remember to adjust the import statement in the__init__.py
file accordingly)- Two lines exceed the maximum line length (see comments)
Default to "mean". | ||
activation_layer: The activation function to use in the model. | ||
Default to "relu". | ||
add_batchnorm_layer: Whether to add a batch normalization layer after |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line too long. See CodeClimate.
skip_readout: Whether to skip the readout layer(s). If `True`, the | ||
output of the last DynEdgeConv block is returned directly. | ||
""" | ||
# Latent feature subset for computing nearest neighbours in ParticleNeT. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line too long. See CodeClimate
src/graphnet/models/gnn/__init__.py
Outdated
@@ -6,3 +6,4 @@ | |||
from .dynedge_kaggle_tito import DynEdgeTITO | |||
from .RNN_tito import RNN_TITO | |||
from .icemix import DeepIce | |||
from .ParticleNeT import ParticleNeT |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
from .particlenet import ParticleNeT
Hello @RasmusOrsoe, I made the changes you commented. Nevertheless, the codeclimate scan shows 6 errors to fix that I think they were previously too. Could you please review this? Thank you! Adding @giogiopg for completeness |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🚀
ParticleNeT model for Jet Tagging with particle cloud-like data used in KM3NeT was successfully implemented in GraphNeT. It uses all the utilities from GraphNeT, mainly DynEdgeConv blocks, to build its architecture. References:
Pull request previously opened by @IvanMM27 with the difference that all the km3net data processing dependencies have been removed.