NN

• callbacks
  • _callbacks
    • CallbackBase
      • CallbackBase.get_default_params()
    • CallbackExecutor
      • CallbackExecutor.execute()
  • early_stopping
    • EarlyStopping
      • EarlyStopping.patience
      • EarlyStopping.min_delta
      • EarlyStopping.get_default_params()
  • model_checkpoint
    • ModelCheckpoint
      • ModelCheckpoint.epoch
      • ModelCheckpoint.max_validation_acc
      • ModelCheckpoint.interval
      • ModelCheckpoint.get_default_params()
      • ModelCheckpoint.save_checkpoint()
  • tensorboard_logger
    • TensorboardLogger
      • TensorboardLogger.epoch
      • TensorboardLogger.writer
      • TensorboardLogger.get_default_params()
  • test_early_stopping
    • test_early_stopping()
• dataloader
  • _dataloader
    • DataLoaderBase
      • DataLoaderBase.collate_fn()
      • DataLoaderBase.get_default_params()
      • DataLoaderBase.get_targets_ids_from_mappings()
    • build_dataloader()
  • simple_dataloader
    • SimpleDataLoader
      • SimpleDataLoader.collate_fn()
      • SimpleDataLoader.get_default_params()
  • simple_metadataloader
    • SimpleMetaDataLoader
      • SimpleMetaDataLoader.collate_fn()
      • SimpleMetaDataLoader.get_default_params()
  • test_simple_dataloader
    • test_metadataloader()
  • test_simple_metadataloader
    • test_metadataloader()
• loss
  • _loss
    • CustomLossBase
      • CustomLossBase.forward()
    • build_loss_fn()
• model
  • _model
    • ModelBase
      • ModelBase.forward()
      • ModelBase.get_default_params()
      • ModelBase.get_predictions()
      • ModelBase.load_weights()
      • ModelBase.save_weights()
    • build_model()
  • sequential_model
    • SequentialModel
      • SequentialModel.forward()
      • SequentialModel.get_default_params()
      • SequentialModel.get_predictions()
      • SequentialModel.make_weights_zero()
  • shap_model
    • CustomShapModel
      • CustomShapModel.forward()
• trainer
  • _trainer
    • TrainerBase
      • TrainerBase.train()
      • TrainerBase.train_one_epoch()
      • TrainerBase.validation()
  • simple_model_trainer
    • SimpleModelTrainer