discotime.models package

Submodules

• discotime.models.components module
  • Block
    • Block.forward()
  • negative_log_likelihood()
• discotime.models.models module
  • LitSurvModule
    • LitSurvModule.configure_optimizers()
    • LitSurvModule.data_cuts
    • LitSurvModule.datamodule
    • LitSurvModule.eval_grid
    • LitSurvModule.forward()
    • LitSurvModule.model
    • LitSurvModule.on_load_checkpoint()
    • LitSurvModule.on_save_checkpoint()
    • LitSurvModule.on_test_epoch_end()
    • LitSurvModule.on_validation_epoch_end()
    • LitSurvModule.setup()
    • LitSurvModule.test_step()
    • LitSurvModule.training_step()
    • LitSurvModule.validation_step()
  • ModelConfig
    • ModelConfig.activation_function
    • ModelConfig.batch_normalization
    • ModelConfig.dropout_rate
    • ModelConfig.evaluation_grid_cuts
    • ModelConfig.evaluation_grid_size
    • ModelConfig.learning_rate
    • ModelConfig.n_hidden_units
    • ModelConfig.n_risks
    • ModelConfig.n_sequential_blocks
    • ModelConfig.n_time_bins
    • ModelConfig.use_skip_connections