graphnet.training package

Modules for training models.

graphnet.training manages model training and experiment logging.

Submodules

• graphnet.training.callbacks module
  • PiecewiseLinearLR
  • ProgressBar
  • GraphnetEarlyStopping
• graphnet.training.labels module
  • Label
  • Direction
  • Track
• graphnet.training.loss_functions module
  • LossFunction
  • MSELoss
  • RMSELoss
  • LogCoshLoss
  • CrossEntropyLoss
  • BinaryCrossEntropyLoss
  • LogCMK
  • VonMisesFisherLoss
  • VonMisesFisher2DLoss
  • EuclideanDistanceLoss
  • VonMisesFisher3DLoss
  • EnsembleLoss
  • RMSEVonMisesFisher3DLoss
• graphnet.training.utils module
  • collate_fn()
  • collator_sequence_buckleting
  • make_dataloader()
  • make_train_validation_dataloader()
  • get_predictions()
  • save_results()
  • save_selection()
• graphnet.training.weight_fitting module
  • WeightFitter
  • Uniform
  • BjoernLow