Index

B

BackwardNRL (class in easy_torch.losses)

BaseNN (class in easy_torch.model) BatchLength (class in easy_torch.metrics)

C

change_all_paddings() (in module easy_torch.torchvision_utils) change_conv_out_features() (in module easy_torch.torchvision_utils) change_fc_out_features() (in module easy_torch.torchvision_utils) change_in_channels() (in module easy_torch.torchvision_utils) complete_prepare_model() (in module easy_torch.preparation) complete_prepare_trainer() (in module easy_torch.preparation) compute() (easy_rec.metrics.F1 method) (easy_rec.metrics.MAP method) (easy_rec.metrics.NDCG method) (easy_rec.metrics.RecMetric method) (easy_torch.metrics.BatchLength method) (easy_torch.metrics.FakeMetric method) (easy_torch.metrics.SoftLabelsAccuracy method)

compute_loss() (easy_torch.model.BaseNN method) compute_metrics() (easy_torch.model.BaseNN method) compute_model_output() (easy_torch.model.BaseNN method) configure_optimizers() (easy_torch.model.BaseNN method) consistency_loss() (easy_torch.losses.NCODLoss method) create_model() (in module easy_torch.process)

D

densify_index_method() (in module easy_rec.data_generation_utils) df_to_sequences() (in module easy_rec.data_generation_utils)

download_dataset() (in module easy_rec.data_generation_utils) DynamicNegatives (class in easy_rec.callbacks)

E

easy_rec.callbacks module easy_rec.data_generation_utils module easy_rec.losses module easy_rec.metrics module easy_torch.callbacks module easy_torch.losses module

easy_torch.metrics module easy_torch.model module easy_torch.preparation module easy_torch.process module easy_torch.torchvision_utils module easy_torch.utils module epsilon (easy_torch.losses.GCELoss attribute)

F

F1 (class in easy_rec.metrics) FakeMetric (class in easy_torch.metrics) FakeMetricCollection (class in easy_torch.metrics) filter_by_frequence() (in module easy_rec.data_generation_utils) filter_ratings() (in module easy_rec.data_generation_utils) forward() (easy_rec.losses.SequentialBCEWithLogitsLoss method) (easy_rec.losses.SequentialBPR method) (easy_rec.losses.SequentialCrossEntropyLoss method) (easy_rec.losses.SequentialGeneralizedBCEWithLogitsLoss method) (easy_torch.losses.BackwardNRL method) (easy_torch.losses.ForwardNRL method) (easy_torch.losses.GCELoss method) (easy_torch.losses.NCODLoss method) (easy_torch.losses.PatriniLoss method) (easy_torch.model.BaseNN method) (easy_torch.model.Identity method) (easy_torch.model.LambdaLayer method)

ForwardNRL (class in easy_torch.losses)

G

gamma_transformation() (easy_rec.losses.SequentialGeneralizedBCEWithLogitsLoss method) GCELoss (class in easy_torch.losses) get_correct_package() (in module easy_torch.preparation) get_function() (in module easy_torch.preparation) get_input_args_kwargs() (easy_torch.model.BaseNN method) get_key_if_dict_and_exists() (easy_torch.model.BaseNN method) get_max_number_of() (in module easy_rec.data_generation_utils)

get_rating_files_per_dataset() (in module easy_rec.data_generation_utils) get_single_callback() (in module easy_torch.preparation) get_single_loss() (in module easy_torch.preparation) get_torchvision_model() (in module easy_torch.model) (in module easy_torch.torchvision_utils) get_torchvision_model_as_decoder() (in module easy_torch.model) get_torchvision_model_split_name() (in module easy_torch.torchvision_utils)

H

handle_FakeMetricCollection() (in module easy_torch.preparation)

I

Identity (class in easy_torch.model) init_param() (easy_torch.losses.NCODLoss method) init_vars() (easy_rec.callbacks.DynamicNegatives method)

invert_layer() (in module easy_torch.torchvision_utils) invert_model() (in module easy_torch.torchvision_utils) items() (easy_torch.utils.RobustModuleDict method)

K

keys() (easy_torch.utils.RobustModuleDict method)

L

LambdaLayer (class in easy_torch.model) load_logs() (in module easy_torch.process) load_model() (in module easy_torch.process)

load_ratings_df() (in module easy_rec.data_generation_utils) load_torchvision_model() (in module easy_torch.model) (in module easy_torch.torchvision_utils) log() (easy_torch.model.BaseNN method)

M

make_fake_class() (in module easy_torch.metrics) MAP (class in easy_rec.metrics) map_dataset_name() (in module easy_rec.data_generation_utils) maybe_preprocess_raw_dataset() (in module easy_rec.data_generation_utils) module easy_rec.callbacks easy_rec.data_generation_utils easy_rec.losses easy_rec.metrics easy_torch.callbacks easy_torch.losses easy_torch.metrics easy_torch.model easy_torch.preparation easy_torch.process easy_torch.torchvision_utils easy_torch.utils

MRR (class in easy_rec.metrics)

N

NCODLoss (class in easy_torch.losses)

NDCG (class in easy_rec.metrics) not_nan_subset() (easy_rec.metrics.RecMetric method)

O

on_epoch_end() (easy_torch.callbacks.TimeCallback method) on_epoch_start() (easy_torch.callbacks.TimeCallback method) on_test_epoch_end() (easy_torch.callbacks.TimeCallback method) on_test_epoch_start() (easy_torch.callbacks.TimeCallback method) on_train_batch_end() (easy_rec.callbacks.DynamicNegatives method) (easy_torch.callbacks.TerminateOnNaNCallback method)

on_train_epoch_end() (easy_rec.callbacks.DynamicNegatives method) (easy_torch.callbacks.TimeCallback method) on_train_epoch_start() (easy_torch.callbacks.TimeCallback method) on_validation_epoch_end() (easy_torch.callbacks.TimeCallback method) on_validation_epoch_start() (easy_torch.callbacks.TemperatureSlowdownCallback method) (easy_torch.callbacks.TimeCallback method)

P

PatriniLoss (class in easy_torch.losses) Precision (class in easy_rec.metrics) PrecisionWithRelevance (class in easy_rec.metrics) predict_step() (easy_torch.model.BaseNN method) prepare_callbacks() (in module easy_torch.preparation) prepare_data_loaders() (in module easy_torch.preparation) prepare_emission_tracker() (in module easy_torch.preparation) prepare_experiment_id() (in module easy_torch.preparation) prepare_flops_profiler() (in module easy_torch.preparation) prepare_logger() (in module easy_torch.preparation) prepare_loss() (in module easy_torch.preparation)

prepare_metrics() (in module easy_torch.preparation) prepare_model() (in module easy_torch.preparation) prepare_optimizer() (in module easy_torch.preparation) prepare_plugins() (in module easy_torch.preparation) prepare_profiler() (in module easy_torch.preparation) prepare_rank_corrections() (in module easy_rec.metrics) prepare_scheduler() (in module easy_torch.preparation) prepare_strategy() (in module easy_torch.preparation) prepare_trainer() (in module easy_torch.preparation) preprocess_dataset() (in module easy_rec.data_generation_utils) print_stats() (in module easy_rec.data_generation_utils)

Q

q (easy_torch.losses.GCELoss attribute)

R

Recall (class in easy_rec.metrics) RecMetric (class in easy_rec.metrics) remove_keys_from_dict() (in module easy_torch.preparation) reset() (easy_rec.metrics.MAP method)

resnet_forward_impl() (in module easy_torch.torchvision_utils) RLS_FRBO (class in easy_rec.metrics) RLS_Jaccard (class in easy_rec.metrics) RLS_RBO (class in easy_rec.metrics) RobustModuleDict (class in easy_torch.utils)

S

SequentialBCEWithLogitsLoss (class in easy_rec.losses) SequentialBPR (class in easy_rec.losses) SequentialCrossEntropyLoss (class in easy_rec.losses) SequentialGeneralizedBCEWithLogitsLoss (class in easy_rec.losses) shutdown_dataloaders_workers() (in module easy_torch.process)

soft_to_hard() (easy_torch.losses.NCODLoss method) SoftLabelsAccuracy (class in easy_torch.metrics) softmax (easy_torch.losses.GCELoss attribute) specific_preprocess() (in module easy_rec.data_generation_utils) split_rec_data() (in module easy_rec.data_generation_utils) step() (easy_torch.model.BaseNN method)

T

TemperatureSlowdownCallback (class in easy_torch.callbacks) TerminateOnNaNCallback (class in easy_torch.callbacks) test_model() (in module easy_torch.process)

test_step() (easy_torch.model.BaseNN method) TimeCallback (class in easy_torch.callbacks) train_model() (in module easy_torch.process) training_step() (easy_torch.model.BaseNN method)

U

update() (easy_rec.metrics.F1 method) (easy_rec.metrics.MAP method) (easy_rec.metrics.MRR method) (easy_rec.metrics.NDCG method) (easy_rec.metrics.Precision method) (easy_rec.metrics.PrecisionWithRelevance method) (easy_rec.metrics.Recall method) (easy_rec.metrics.RLS_FRBO method) (easy_rec.metrics.RLS_Jaccard method) (easy_rec.metrics.RLS_RBO method) (easy_torch.metrics.BatchLength method) (easy_torch.metrics.FakeMetric method) (easy_torch.metrics.SoftLabelsAccuracy method) (easy_torch.utils.RobustModuleDict method)

V

validate_model() (in module easy_torch.process) validation_step() (easy_torch.model.BaseNN method)

values() (easy_torch.utils.RobustModuleDict method) video_resnet_forward() (in module easy_torch.torchvision_utils)