graphnet.models.task package

Physics task-specific modules to be used as model “read-outs”.

Submodules

• graphnet.models.task.classification module
  • MulticlassClassificationTask
  • BinaryClassificationTask
  • BinaryClassificationTaskLogits
• graphnet.models.task.reconstruction module
  • AzimuthReconstructionWithKappa
  • AzimuthReconstruction
  • DirectionReconstructionWithKappa
  • ZenithReconstruction
  • ZenithReconstructionWithKappa
  • EnergyReconstruction
  • EnergyReconstructionWithPower
  • EnergyTCReconstruction
  • EnergyReconstructionWithUncertainty
  • VertexReconstruction
  • PositionReconstruction
  • TimeReconstruction
  • InelasticityReconstruction
• graphnet.models.task.task module
  • Task
  • LearnedTask
  • StandardLearnedTask
  • IdentityTask
  • StandardFlowTask