Index

A

aliases (ctlearn.tools.train_model.TrainCTLearnModel attribute) architecture (ctlearn.core.model.ResNet attribute) (ctlearn.core.model.SingleCNN attribute)

attention_mechanism (ctlearn.core.model.CTLearnModel attribute) attention_reduction_ratio (ctlearn.core.model.CTLearnModel attribute)

B

batch_size (ctlearn.tools.train_model.TrainCTLearnModel attribute) batchnorm (ctlearn.core.model.SingleCNN attribute)

bottleneck_filters (ctlearn.core.model.SingleCNN attribute) build_fully_connect_head() (in module ctlearn.core.model)

C

channel_squeeze_excite_block() (in module ctlearn.core.attention) classes (ctlearn.tools.train_model.TrainCTLearnModel attribute) ctlearn module ctlearn.core.attention module

ctlearn.core.loader module ctlearn.core.model module ctlearn.tools.train_model module CTLearnModel (class in ctlearn.core.model)

D

description (ctlearn.tools.train_model.TrainCTLearnModel attribute) dl1dh_reader_type (ctlearn.tools.train_model.TrainCTLearnModel attribute)

DLDataLoader (class in ctlearn.core.loader) dual_squeeze_excite_block() (in module ctlearn.core.attention)

E

early_stopping (ctlearn.tools.train_model.TrainCTLearnModel attribute)

examples (ctlearn.tools.train_model.TrainCTLearnModel attribute)

F

file_pattern_background (ctlearn.tools.train_model.TrainCTLearnModel attribute)

file_pattern_signal (ctlearn.tools.train_model.TrainCTLearnModel attribute) finish() (ctlearn.tools.train_model.TrainCTLearnModel method)

H

head_activation_function (ctlearn.core.model.CTLearnModel attribute)

head_layers (ctlearn.core.model.CTLearnModel attribute)

I

init_layer (ctlearn.core.model.ResNet attribute) init_max_pool (ctlearn.core.model.ResNet attribute)

init_padding (ctlearn.core.model.CTLearnModel attribute) input_dir_background (ctlearn.tools.train_model.TrainCTLearnModel attribute) input_dir_signal (ctlearn.tools.train_model.TrainCTLearnModel attribute)

L

load_model_from (ctlearn.core.model.LoadedModel attribute)

LoadedModel (class in ctlearn.core.model) lr_reducing (ctlearn.tools.train_model.TrainCTLearnModel attribute)

M

main() (in module ctlearn.tools.train_model) model_type (ctlearn.tools.train_model.TrainCTLearnModel attribute) module ctlearn ctlearn.core.attention ctlearn.core.loader ctlearn.core.model ctlearn.tools.train_model

N

n_epochs (ctlearn.tools.train_model.TrainCTLearnModel attribute) name (ctlearn.core.model.ResNet attribute) (ctlearn.core.model.SingleCNN attribute) (ctlearn.tools.train_model.TrainCTLearnModel attribute)

O

on_epoch_end() (ctlearn.core.loader.DLDataLoader method) optimizer (ctlearn.tools.train_model.TrainCTLearnModel attribute)

output_dir (ctlearn.tools.train_model.TrainCTLearnModel attribute) overwrite_head (ctlearn.core.model.LoadedModel attribute)

P

pooling_parameters (ctlearn.core.model.SingleCNN attribute)

pooling_type (ctlearn.core.model.SingleCNN attribute)

R

random_seed (ctlearn.tools.train_model.TrainCTLearnModel attribute) reco_tasks (ctlearn.tools.train_model.TrainCTLearnModel attribute)

residual_block_type (ctlearn.core.model.ResNet attribute) ResNet (class in ctlearn.core.model)

S

save_best_validation_only (ctlearn.tools.train_model.TrainCTLearnModel attribute) save_onnx (ctlearn.tools.train_model.TrainCTLearnModel attribute) setup() (ctlearn.tools.train_model.TrainCTLearnModel method) SingleCNN (class in ctlearn.core.model)

sort_by_intensity (ctlearn.tools.train_model.TrainCTLearnModel attribute) spatial_squeeze_excite_block() (in module ctlearn.core.attention) stack_telescope_images (ctlearn.tools.train_model.TrainCTLearnModel attribute) start() (ctlearn.tools.train_model.TrainCTLearnModel method)

T

trainable_backbone (ctlearn.core.model.LoadedModel attribute)

TrainCTLearnModel (class in ctlearn.tools.train_model)

V

validation_split (ctlearn.tools.train_model.TrainCTLearnModel attribute)