Index

A

access_by_name() (in module ebes.pipeline.utils) AllHiddenMean (class in ebes.model.agg) assign_by_name() (in module ebes.pipeline.utils)

Attention (class in ebes.model.PrimeNet.modules) attention() (ebes.model.mtand.MultiTimeAttention method) (ebes.model.PrimeNet.models.MultiTimeAttention method)

B

backward (ebes.data.batch_tfs.ForwardFillNans attribute) BaseAgg (class in ebes.model.agg) BaseModel (class in ebes.model.basemodel) BaseSeq2Seq (class in ebes.model.seq2seq) Batch (class in ebes.types) Batch2Seq (class in ebes.model.preprocess)

batch_transforms (ebes.data.loading.SequenceCollator attribute) BatchTransform (class in ebes.data.batch_tfs) BertInterpHead (class in ebes.model.PrimeNet.models) BertPooler (class in ebes.model.PrimeNet.models) best_checkpoint() (ebes.trainer.Trainer method) build_loaders() (in module ebes.data.utils) build_model() (in module ebes.model.utils)

C

cat_cardinalities (ebes.data.loading.SequenceCollator attribute) cat_features (ebes.types.Batch attribute) cat_features_names (ebes.types.Batch attribute) cat_mask (ebes.types.Batch attribute) CatToNum (class in ebes.data.batch_tfs) cl_loss (ebes.types.PrimeNetReturn attribute) clus_labels (ebes.types.NHReturn attribute) (ebes.types.NHSeq attribute) clustering_loss (ebes.types.NHReturn attribute) (ebes.types.NHSeq attribute) cnt_max (ebes.data.batch_tfs.RandomSlices attribute)

cnt_min (ebes.data.batch_tfs.RandomSlices attribute) compute() (ebes.metrics.custom.LoggingMetric method) (ebes.metrics.custom.MultiLabelMeanAUROC method) (ebes.metrics.custom.NegRootMeanSquaredError method) (ebes.metrics.custom.PrimeNetAccuracy method) (ebes.metrics.neural_hawkes.NHEventTypeAccuracy method) compute_metrics() (ebes.trainer.Trainer method) ContrastiveLoss (class in ebes.losses.contrastive) ContrastiveTarget (class in ebes.data.batch_tfs) correct_num (ebes.types.PrimeNetReturn attribute) create_block() (in module ebes.model.mamba.mamba_es)

D

DatetimeToFloat (class in ebes.data.batch_tfs) dense_sampling_bound (ebes.data.batch_tfs.PrimeNetSampler attribute)

device (ebes.trainer.Trainer property) do_n_runs() (ebes.pipeline.base_runner.Runner method)

E

ebes module ebes.data module ebes.data.accessors module ebes.data.batch_tfs module ebes.data.datasets module ebes.data.loading module ebes.data.utils module ebes.losses module ebes.losses.base module ebes.losses.contrastive module ebes.losses.multi_label module ebes.losses.neural_hawkes module ebes.metrics module ebes.metrics.custom module ebes.metrics.neural_hawkes module ebes.model module ebes.model.agg module ebes.model.basemodel module ebes.model.mamba module ebes.model.mamba.mamba_es module ebes.model.mtand module

ebes.model.preprocess module ebes.model.PrimeNet module ebes.model.PrimeNet.learn_time_emb module ebes.model.PrimeNet.models module ebes.model.PrimeNet.modules module ebes.model.seq2seq module ebes.model.utils module ebes.pipeline module ebes.pipeline.base_runner module ebes.pipeline.runners module ebes.pipeline.runners.evaluate module ebes.pipeline.runners.primenet module ebes.pipeline.runners.supervised module ebes.pipeline.runners.unsupervised module ebes.pipeline.utils module ebes.trainer module ebes.types module ebes.utils module ebes.utils.general module ebes.utils.reproduce module encode() (ebes.model.PrimeNet.models.SwitchTimeBERT method) (ebes.model.PrimeNet.models.TimeBERT method) EvalRunner (class in ebes.pipeline.runners.evaluate)

F

fill_value (ebes.data.batch_tfs.FillNans attribute) FillNans (class in ebes.data.batch_tfs) forward() (ebes.losses.base.ModelLoss method) (ebes.losses.contrastive.ContrastiveLoss method) (ebes.losses.contrastive.InfoNCELoss method) (ebes.losses.multi_label.MultiLabelBinaryCrossEntropyLoss method) (ebes.losses.neural_hawkes.NHLoss method) (ebes.model.agg.AllHiddenMean method) (ebes.model.agg.BaseAgg method) (ebes.model.agg.TakeLastHidden method) (ebes.model.agg.ToTensor method) (ebes.model.agg.ValidHiddenMean method) (ebes.model.mamba.mamba_es.MambaModel method) (ebes.model.mamba.mamba_es.MixerModel method) (ebes.model.mtand.MTAND method) (ebes.model.mtand.MultiTimeAttention method) (ebes.model.preprocess.Batch2Seq method) (ebes.model.preprocess.SeqBatchNorm method) (ebes.model.PrimeNet.learn_time_emb.MultiTimeAttention method) (ebes.model.PrimeNet.learn_time_emb.Similarity method) (ebes.model.PrimeNet.learn_time_emb.TimeBERT method) (ebes.model.PrimeNet.models.BertInterpHead method) (ebes.model.PrimeNet.models.BertPooler method) (ebes.model.PrimeNet.models.MultiTimeAttention method) (ebes.model.PrimeNet.models.Similarity method) (ebes.model.PrimeNet.models.SwitchTimeBERT method) (ebes.model.PrimeNet.models.TimeBERT method) (ebes.model.PrimeNet.models.TimeBERTForMultiTask method) (ebes.model.PrimeNet.modules.Attention method) (ebes.model.PrimeNet.modules.GELU method) (ebes.model.PrimeNet.modules.LayerNorm method) (ebes.model.PrimeNet.modules.MultiHeadedAttention method) (ebes.model.PrimeNet.modules.OutputLayer method) (ebes.model.PrimeNet.modules.PositionwiseFeedForward method) (ebes.model.PrimeNet.modules.SublayerConnection method) (ebes.model.PrimeNet.modules.SwitchTransformerBlock method) (ebes.model.PrimeNet.modules.TransformerBlock method) (ebes.model.seq2seq.BaseSeq2Seq method) (ebes.model.seq2seq.GRU method) (ebes.model.seq2seq.PositionalEncoding method) (ebes.model.seq2seq.Projection method) (ebes.model.seq2seq.Transformer method) (ebes.model.utils.FrozenModel method)

ForwardFillNans (class in ebes.data.batch_tfs) FrozenModel (class in ebes.model.utils)

G

GELU (class in ebes.model.PrimeNet.modules) get_accessor() (in module ebes.data.utils) get_collator() (in module ebes.data.utils) get_dict_from_trial_params() (in module ebes.pipeline.utils) get_global_state() (in module ebes.utils.reproduce) get_loader() (in module ebes.data.utils) get_loss() (in module ebes.pipeline.utils) get_metrics() (in module ebes.pipeline.utils) get_model() (ebes.model.basemodel.BaseModel static method) get_optimizer() (in module ebes.pipeline.utils)

get_pairs() (ebes.losses.contrastive.HardNegativePairSelector method) (ebes.losses.contrastive.PairSelector method) get_pe() (ebes.model.seq2seq.PositionalEncoding method) get_runner() (ebes.pipeline.base_runner.Runner static method) get_scheduler() (in module ebes.pipeline.utils) get_split() (ebes.data.accessors.InMemoryPandasDataAccessor method) (ebes.data.accessors.PandasDataAccessor method) get_unique_folder_suffix() (in module ebes.pipeline.utils) grad_norm() (in module ebes.utils.general) GRU (class in ebes.model.seq2seq)

H

HardNegativePairSelector (class in ebes.losses.contrastive)

I

index (ebes.types.Batch attribute) index_name (ebes.data.loading.SequenceCollator attribute)

InfoNCELoss (class in ebes.losses.contrastive) InMemoryPandasDataAccessor (class in ebes.data.accessors) isnan() (in module ebes.model.PrimeNet.models)

K

keep_last (ebes.data.batch_tfs.RandomEventsPermutation attribute)

L

LayerNorm (class in ebes.model.PrimeNet.modules) learn_time_embedding() (ebes.model.mtand.MTAND method) (ebes.model.PrimeNet.learn_time_emb.TimeBERT method) (ebes.model.PrimeNet.models.SwitchTimeBERT method) (ebes.model.PrimeNet.models.TimeBERT method) len_sampling_bound (ebes.data.batch_tfs.PrimeNetSampler attribute) lengths (ebes.types.Batch attribute) (ebes.types.NHReturn attribute) (ebes.types.Seq attribute) load_best_model() (ebes.trainer.Trainer method)

load_ckpt() (ebes.trainer.Trainer method) LoadTime (class in ebes.utils.general) loc (ebes.data.batch_tfs.DatetimeToFloat attribute) (ebes.data.batch_tfs.Rescale attribute) (ebes.data.batch_tfs.RescaleTime attribute) log_to_file() (in module ebes.utils.general) Logarithm (class in ebes.data.batch_tfs) LoggingMetric (class in ebes.metrics.custom) loss (ebes.types.PrimeNetReturn attribute) lr_scheduler (ebes.trainer.Trainer property)

M

MambaModel (class in ebes.model.mamba.mamba_es) mask_ratio_per_seg (ebes.data.batch_tfs.PrimeNetSampler attribute) masks (ebes.types.Seq attribute) MaskValid (class in ebes.data.batch_tfs) max_seq_len (ebes.data.loading.SequenceCollator attribute) merge_state() (ebes.metrics.custom.LoggingMetric method) (ebes.metrics.custom.PrimeNetAccuracy method) (ebes.metrics.neural_hawkes.NHEventTypeAccuracy method) MixerModel (class in ebes.model.mamba.mamba_es) MLEM_reconstruction_loss (class in ebes.metrics.custom) MLEM_sparcity_loss (class in ebes.metrics.custom) MLEM_total_CE_loss (class in ebes.metrics.custom) MLEM_total_mse_loss (class in ebes.metrics.custom) model (ebes.trainer.Trainer property) ModelLoss (class in ebes.losses.base) module ebes ebes.data ebes.data.accessors ebes.data.batch_tfs ebes.data.datasets ebes.data.loading ebes.data.utils ebes.losses ebes.losses.base ebes.losses.contrastive ebes.losses.multi_label ebes.losses.neural_hawkes ebes.metrics ebes.metrics.custom ebes.metrics.neural_hawkes ebes.model ebes.model.agg ebes.model.basemodel ebes.model.mamba ebes.model.mamba.mamba_es ebes.model.mtand ebes.model.preprocess ebes.model.PrimeNet ebes.model.PrimeNet.learn_time_emb ebes.model.PrimeNet.models ebes.model.PrimeNet.modules ebes.model.seq2seq ebes.model.utils ebes.pipeline ebes.pipeline.base_runner ebes.pipeline.runners ebes.pipeline.runners.evaluate ebes.pipeline.runners.primenet ebes.pipeline.runners.supervised ebes.pipeline.runners.unsupervised ebes.pipeline.utils ebes.trainer ebes.types ebes.utils ebes.utils.general ebes.utils.reproduce

mse_loss (ebes.types.PrimeNetReturn attribute) MTAND (class in ebes.model.mtand) MultiHeadedAttention (class in ebes.model.PrimeNet.modules) MultiLabelBinaryCrossEntropyLoss (class in ebes.losses.multi_label) MultiLabelMeanAUROC (class in ebes.metrics.custom) MultiTimeAttention (class in ebes.model.mtand) (class in ebes.model.PrimeNet.learn_time_emb) (class in ebes.model.PrimeNet.models)

N

name (ebes.data.batch_tfs.Rescale attribute) names (ebes.data.batch_tfs.Logarithm attribute) NegRootMeanSquaredError (class in ebes.metrics.custom) NHEventLogIntensity (class in ebes.metrics.neural_hawkes) NHEventTypeAccuracy (class in ebes.metrics.neural_hawkes) NHLL (class in ebes.metrics.neural_hawkes) NHLoss (class in ebes.losses.neural_hawkes)

NHNegNonEventIntensity (class in ebes.metrics.neural_hawkes) NHReturn (class in ebes.types) NHSeq (class in ebes.types) non_event_intensity (ebes.types.NHReturn attribute) num_features (ebes.types.Batch attribute) num_features_names (ebes.types.Batch attribute) num_mask (ebes.types.Batch attribute) num_names (ebes.data.loading.SequenceCollator attribute)

O

optimizer (ebes.trainer.Trainer property) optuna_df() (in module ebes.pipeline.utils) outer_pairwise_distance() (in module ebes.losses.contrastive) output_dim (ebes.model.mamba.mamba_es.MambaModel property) (ebes.model.mtand.MTAND property) (ebes.model.preprocess.Batch2Seq property) (ebes.model.seq2seq.GRU property) (ebes.model.seq2seq.Projection property) (ebes.model.seq2seq.Transformer property)

OutputLayer (class in ebes.model.PrimeNet.modules)

P

padding_type (ebes.data.loading.SequenceCollator attribute) PairSelector (class in ebes.losses.contrastive) PandasDataAccessor (class in ebes.data.accessors) param_grid() (ebes.pipeline.base_runner.Runner method) (ebes.pipeline.runners.evaluate.EvalRunner method) (ebes.pipeline.runners.primenet.PrimeNetRunner method) (ebes.pipeline.runners.supervised.SupervisedRunner method) (ebes.pipeline.runners.unsupervised.UnsupervisedRunner method) parse_n_runs() (in module ebes.pipeline.utils) pipeline() (ebes.pipeline.base_runner.Runner method) (ebes.pipeline.runners.evaluate.EvalRunner method) (ebes.pipeline.runners.primenet.PrimeNetRunner method) (ebes.pipeline.runners.supervised.SupervisedRunner method) (ebes.pipeline.runners.unsupervised.UnsupervisedRunner method)

pop_target() (ebes.types.Batch method) PositionalEncoding (class in ebes.model.seq2seq) PositionwiseFeedForward (class in ebes.model.PrimeNet.modules) pre_event_intensities_of_gt (ebes.types.NHReturn attribute) pred_labels (ebes.types.NHReturn attribute) pretrain_tasks (ebes.data.batch_tfs.PrimeNetSampler attribute) PrimeNetAccuracy (class in ebes.metrics.custom) PrimeNetReturn (class in ebes.types) PrimeNetRunner (class in ebes.pipeline.runners.primenet) PrimeNetSampler (class in ebes.data.batch_tfs) process_type (ebes.data.batch_tfs.TimeToFeatures attribute) Projection (class in ebes.model.seq2seq)

R

RandomEventsPermutation (class in ebes.data.batch_tfs) RandomSlices (class in ebes.data.batch_tfs) RandomTime (class in ebes.data.batch_tfs) Rescale (class in ebes.data.batch_tfs) RescaleTime (class in ebes.data.batch_tfs)

run() (ebes.pipeline.base_runner.Runner method) (ebes.trainer.Trainer method) run_name (ebes.trainer.Trainer property) run_optuna() (ebes.pipeline.base_runner.Runner method) Runner (class in ebes.pipeline.base_runner)

S

save_ckpt() (ebes.trainer.Trainer method) scale (ebes.data.batch_tfs.DatetimeToFloat attribute) (ebes.data.batch_tfs.Rescale attribute) (ebes.data.batch_tfs.RescaleTime attribute) seed (ebes.data.batch_tfs.RandomSlices attribute) seed_everything() (in module ebes.utils.reproduce) segment_num (ebes.data.batch_tfs.PrimeNetSampler attribute) Seq (class in ebes.types) SeqBatchNorm (class in ebes.model.preprocess) SequenceCollator (class in ebes.data.loading) series() (in module ebes.data.datasets) SeriesDataset (class in ebes.data.datasets)

set_global_state() (in module ebes.utils.reproduce) set_start_method() (in module ebes.pipeline.utils) set_start_method_as_fork() (in module ebes.pipeline.utils) short_seq_crop_rate (ebes.data.batch_tfs.RandomSlices attribute) Similarity (class in ebes.model.PrimeNet.learn_time_emb) (class in ebes.model.PrimeNet.models) SizedSeriesDataset (class in ebes.data.datasets) split_count (ebes.data.batch_tfs.RandomSlices attribute) SublayerConnection (class in ebes.model.PrimeNet.modules) suggest_conf() (in module ebes.pipeline.utils) SupervisedRunner (class in ebes.pipeline.runners.supervised) SwitchTimeBERT (class in ebes.model.PrimeNet.models) SwitchTransformerBlock (class in ebes.model.PrimeNet.modules)

T

TakeLastHidden (class in ebes.model.agg) target (ebes.types.Batch attribute) target_name (ebes.data.loading.SequenceCollator attribute) TargetToLong (class in ebes.data.batch_tfs) time (ebes.types.Batch attribute) (ebes.types.Seq attribute) time_embedding() (ebes.model.PrimeNet.models.SwitchTimeBERT method) (ebes.model.PrimeNet.models.TimeBERT method) time_name (ebes.data.batch_tfs.TimeToFeatures attribute) (ebes.data.loading.SequenceCollator attribute) TimeBERT (class in ebes.model.PrimeNet.learn_time_emb) (class in ebes.model.PrimeNet.models) TimeBERTConfig (class in ebes.model.PrimeNet.models) TimeBERTForMultiTask (class in ebes.model.PrimeNet.models)

TimeToFeatures (class in ebes.data.batch_tfs) to() (ebes.types.Batch method) (ebes.types.NHReturn method) (ebes.types.PrimeNetReturn method) (ebes.types.Seq method) tokens (ebes.types.Seq attribute) total_num (ebes.types.PrimeNetReturn attribute) ToTensor (class in ebes.model.agg) train() (ebes.model.utils.FrozenModel method) (ebes.trainer.Trainer method) train_loader (ebes.trainer.Trainer property) Trainer (class in ebes.trainer) Transformer (class in ebes.model.seq2seq) TransformerBlock (class in ebes.model.PrimeNet.modules)

U

UnsqueezeTarget (class in ebes.data.batch_tfs) UnsupervisedRunner (class in ebes.pipeline.runners.unsupervised) update() (ebes.metrics.custom.LoggingMetric method) (ebes.metrics.custom.MultiLabelMeanAUROC method) (ebes.metrics.custom.NegRootMeanSquaredError method) (ebes.metrics.custom.PrimeNetAccuracy method) (ebes.metrics.neural_hawkes.NHEventLogIntensity method) (ebes.metrics.neural_hawkes.NHEventTypeAccuracy method) (ebes.metrics.neural_hawkes.NHLL method) (ebes.metrics.neural_hawkes.NHNegNonEventIntensity method)

V

val_loader (ebes.trainer.Trainer property)

validate() (ebes.trainer.Trainer method) ValidHiddenMean (class in ebes.model.agg)