Index

A

accimage_loader() (in module libauc.datasets.folder) Adam (class in libauc.optimizers.adam) AdamSpace (class in libauc.trainer.config.spaces) AdamW (class in libauc.optimizers.adamw) add_callback() (CallbackHandler method) (Trainer method) adjust_gamma() (GCLoss method) (GCLoss_v1 method) (GCLoss_v2 method)

adjust_lr() (in module libauc.utils.paper_utils) ap_at_k() (in module libauc.metrics.metrics_k) APLossSpace (class in libauc.trainer.config.spaces) as_array() (CIFAR10 method) AtomEncoder (class in libauc.models.gnn) auc_prc_score() (in module libauc.metrics.metrics) auc_roc_score() (in module libauc.metrics.metrics) AUCMLoss (class in libauc.losses.auc) AUCMLossSpace (class in libauc.trainer.config.spaces) AveragePrecisionLoss (class in libauc.losses.auc)

B

barrier_hinge_loss() (in module libauc.losses.surrogate) base_folder (CIFAR10 attribute) (CIFAR100 attribute)

batch_to_gpu() (in module libauc.utils.paper_utils) BCELossSpace (class in libauc.trainer.config.spaces) BondEncoder (class in libauc.models.gnn) build_metric() (in module libauc.trainer.helpers)

C

cal_ideal_dcg() (in module libauc.datasets.movielens) callback_list (CallbackHandler property) CallbackHandler (class in libauc.trainer.core.callbacks) CAT_VS_DOG() (in module libauc.datasets.cat_vs_dog) check_array_shape() (in module libauc.metrics.metrics_k) (in module libauc.utils.utils) check_array_type() (in module libauc.metrics.metrics_k) (in module libauc.utils.utils) check_class_labels() (in module libauc.utils.utils) check_imbalance_ratio() (in module libauc.utils.utils) check_tensor_shape() (in module libauc.utils.utils) ChemicalDataset (class in libauc.trainer.data.datasets)

CIFAR10 (class in libauc.datasets.cifar) CIFAR100 (class in libauc.datasets.cifar) CLICallback (class in libauc.trainer.core.callbacks) collate_batch() (MoiveLens method) (moivelens_trainset method) collate_fn() (in module libauc.utils.paper_utils) CompositionalAUCLoss (class in libauc.losses.auc) CompositionalAUCLossSpace (class in libauc.trainer.config.spaces) concat_all_gather() (in module libauc.utils.utils) ControlledDataSampler (class in libauc.sampler.sampler) CosineLRScheduler (class in libauc.utils.paper_utils) CrossEntropyLoss (class in libauc.losses.losses)

D

DatasetFolder (class in libauc.datasets.folder) DeeperGCN (class in libauc.models.gnn) default_loader() (in module libauc.datasets.folder) DefaultCallback (class in libauc.trainer.core.callbacks) DenseNet (class in libauc.models.densenet) densenet121() (in module libauc.models.densenet) densenet161() (in module libauc.models.densenet) densenet169() (in module libauc.models.densenet)

densenet201() (in module libauc.models.densenet) detshuffle2 (class in libauc.datasets.webdataset) df_to_dict() (in module libauc.datasets.movielens) DistributedDualSampler (class in libauc.sampler.sampler) DistributedTriSampler (class in libauc.sampler.sampler) download() (CIFAR10 method) download_dataset() (in module libauc.datasets.movielens) DualSampler (class in libauc.sampler.sampler)

E

evaluate() (GNNTrainer method) (in module libauc.utils.paper_utils) (Trainer method)

evaluate_loop() (Trainer method) evaluate_method() (in module libauc.utils.paper_utils) evaluator() (in module libauc.metrics.metrics) extra_repr() (CIFAR10 method)

F

filename (CIFAR10 attribute) (CIFAR100 attribute) filter_no_caption_or_no_image() (in module libauc.datasets.webdataset)

find_classes() (DatasetFolder method) (in module libauc.datasets.folder) FocalLoss (class in libauc.losses.losses) format_metric() (in module libauc.utils.paper_utils)

G

GAT (class in libauc.models.gnn) GCLoss (class in libauc.losses.contrastive) GCLoss_v1 (class in libauc.losses.contrastive) GCLoss_v2 (class in libauc.losses.contrastive) GCN (class in libauc.models.gnn) get_augmentations_v1() (in module libauc.datasets.melanoma) get_batch() (MoiveLens method) (moivelens_evalset method) get_cycle_length() (CosineLRScheduler method) get_epoch_values() (CosineLRScheduler method) (Scheduler method) get_latest_checkpoint() (Trainer method)

get_num_televant_pairs() (MoiveLens method) (moivelens_trainset method) get_surrogate_loss() (in module libauc.losses.surrogate) get_time() (in module libauc.utils.utils) get_update_values() (CosineLRScheduler method) (Scheduler method) get_value() (SharedEpoch method) GIN (class in libauc.models.gnn) GINE (class in libauc.models.gnn) GNNTrainer (class in libauc.trainer.core.gnn_trainer) GraphDataset (class in libauc.trainer.data.datasets) GraphSAGE (class in libauc.models.gnn) group_by_keys_nothrow() (in module libauc.datasets.webdataset)

H

has_file_allowed_extension() (in module libauc.datasets.folder)

hinge_loss() (in module libauc.losses.surrogate)

I

ImageDataset (class in libauc.trainer.data.datasets) ImageFolder (class in libauc.datasets.folder) ImbalancedDataGenerator (class in libauc.utils.utils) IndexedDataset (class in libauc.trainer.data.datasets) init_conv() (GAT method) (GCN method) (GIN method) (GINE method) (GraphSAGE method) (MPNN method) (PNA method)

init_weights() (NeuMF static method) is_distributed() (in module libauc.optimizers.pesg) (in module libauc.utils.utils) is_image_file() (in module libauc.datasets.folder) iSogCLR (class in libauc.optimizers.isogclr)

L

LARS (class in libauc.optimizers.lars) libauc module libauc.datasets.cat_vs_dog module libauc.datasets.cifar module libauc.datasets.folder module libauc.datasets.melanoma module libauc.datasets.movielens module libauc.datasets.stl10 module libauc.datasets.webdataset module libauc.losses.auc module libauc.losses.contrastive module libauc.losses.losses module libauc.losses.mil module libauc.losses.ranking module libauc.losses.surrogate module libauc.metrics.metrics module libauc.metrics.metrics_k module libauc.models.densenet module libauc.models.gnn module libauc.models.neumf module libauc.models.perceptron module libauc.models.resnet module libauc.models.resnet_cifar module libauc.optimizers.adam module libauc.optimizers.adamw module libauc.optimizers.isogclr module libauc.optimizers.lars module libauc.optimizers.midam module libauc.optimizers.pdsca module libauc.optimizers.pesg module libauc.optimizers.sgd module libauc.optimizers.soap module libauc.optimizers.sogclr module

libauc.optimizers.song module libauc.optimizers.sopa module libauc.optimizers.sopa_s module libauc.optimizers.sota_s module libauc.sampler.sampler module libauc.trainer module libauc.trainer.config module libauc.trainer.config.args module libauc.trainer.config.spaces module libauc.trainer.core module libauc.trainer.core.callbacks module libauc.trainer.core.gnn_trainer module libauc.trainer.core.trainer module libauc.trainer.data module libauc.trainer.data.datasets module libauc.trainer.helpers module libauc.utils.paper_utils module libauc.utils.utils module ListwiseCELoss (class in libauc.losses.ranking) load_checkpoint() (Trainer method) load_data() (in module libauc.datasets.cat_vs_dog) load_dataset() (in module libauc.trainer.data.datasets) load_file() (in module libauc.datasets.stl10) load_model() (NeuMF method) load_state_dict() (AUCMLoss method) (CompositionalAUCLoss method) (PDSCA method) (PESG method) (Scheduler method) log_and_continue() (in module libauc.datasets.webdataset) logistic_loss() (in module libauc.losses.surrogate) loss (AdamSpace attribute) (APLossSpace attribute) (AUCMLossSpace attribute) (BCELossSpace attribute) (CompositionalAUCLossSpace attribute) (mAPLossSpace attribute) (MultiLabelAUCMLossSpace attribute) (MultiLabelpAUC_CVaR_LossSpace attribute) (MultiLabelpAUC_DRO_LossSpace attribute) (MultiLabeltpAUC_KL_LossSpace attribute) (NDCGLossSpace attribute) (pAUC_CVaR_LossSpace attribute) (pAUC_DRO_LossSpace attribute) (SGDSpace attribute) (tpAUC_CVaR_lossSpace attribute) (tpAUC_KL_LossSpace attribute)

M

make_dataset() (DatasetFolder static method) (in module libauc.datasets.folder) map_at_k() (in module libauc.metrics.metrics_k) mAPLossSpace (class in libauc.trainer.config.spaces) mean() (CompositionalAUCLoss method) (MultiLabelAUCMLoss method) meanAveragePrecisionLoss (class in libauc.losses.auc) MedicalImageCSVDataset (class in libauc.trainer.data.datasets) Melanoma (class in libauc.datasets.melanoma) meta (CIFAR10 attribute) (CIFAR100 attribute) MIDAM (class in libauc.optimizers.midam) MIDAM_attention_pooling_loss (class in libauc.losses.mil) MIDAM_softmax_pooling_loss (class in libauc.losses.mil) MIDAMLoss (class in libauc.losses.mil) MIL_aggregation() (in module libauc.utils.paper_utils) MIL_evaluate_auc() (in module libauc.utils.paper_utils) MIL_sampling() (in module libauc.utils.paper_utils) MLP (class in libauc.models.perceptron) module libauc libauc.datasets.cat_vs_dog libauc.datasets.cifar libauc.datasets.folder libauc.datasets.melanoma libauc.datasets.movielens libauc.datasets.stl10 libauc.datasets.webdataset libauc.losses.auc libauc.losses.contrastive libauc.losses.losses libauc.losses.mil libauc.losses.ranking libauc.losses.surrogate libauc.metrics.metrics libauc.metrics.metrics_k libauc.models.densenet libauc.models.gnn libauc.models.neumf libauc.models.perceptron libauc.models.resnet libauc.models.resnet_cifar libauc.optimizers.adam libauc.optimizers.adamw libauc.optimizers.isogclr libauc.optimizers.lars libauc.optimizers.midam libauc.optimizers.pdsca libauc.optimizers.pesg libauc.optimizers.sgd libauc.optimizers.soap libauc.optimizers.sogclr libauc.optimizers.song libauc.optimizers.sopa libauc.optimizers.sopa_s libauc.optimizers.sota_s libauc.sampler.sampler libauc.trainer libauc.trainer.config libauc.trainer.config.args libauc.trainer.config.spaces libauc.trainer.core libauc.trainer.core.callbacks libauc.trainer.core.gnn_trainer libauc.trainer.core.trainer libauc.trainer.data libauc.trainer.data.datasets libauc.trainer.helpers libauc.utils.paper_utils libauc.utils.utils

MoiveLens (class in libauc.datasets.movielens) moivelens_evalset (class in libauc.datasets.movielens) moivelens_trainset (class in libauc.datasets.movielens) MPNN (class in libauc.models.gnn) MultiLabelAUCMLoss (class in libauc.losses.auc) MultiLabelAUCMLossSpace (class in libauc.trainer.config.spaces) MultiLabelpAUC_CVaR_LossSpace (class in libauc.trainer.config.spaces) MultiLabelpAUC_DRO_LossSpace (class in libauc.trainer.config.spaces) MultiLabelpAUCLoss (class in libauc.losses.auc) MultiLabeltpAUC_KL_LossSpace (class in libauc.trainer.config.spaces)

N

ndcg_at_k() (in module libauc.metrics.metrics_k) NDCGLoss (class in libauc.losses.ranking)

NDCGLossSpace (class in libauc.trainer.config.spaces) NeuMF (class in libauc.models.neumf)

O

on_epoch_begin() (CallbackHandler method) (CLICallback method) (DefaultCallback method) (TrainerCallback method) on_epoch_end() (CallbackHandler method) (CLICallback method) (DefaultCallback method) (TrainerCallback method) on_evaluate() (CallbackHandler method) (TrainerCallback method) on_init_end() (CallbackHandler method) (TrainerCallback method) on_log() (CallbackHandler method) (TrainerCallback method) on_predict() (CallbackHandler method) (TrainerCallback method) on_prediction_step() (CallbackHandler method) (TrainerCallback method) on_save() (CallbackHandler method) (TrainerCallback method) on_step_begin() (CallbackHandler method) (TrainerCallback method) on_step_end() (CallbackHandler method) (CLICallback method) (DefaultCallback method) (TrainerCallback method)

on_substep_end() (CallbackHandler method) (TrainerCallback method) on_train_begin() (CallbackHandler method) (CLICallback method) (TrainerCallback method) on_train_end() (CallbackHandler method) (CLICallback method) (TrainerCallback method) optim_step (PDSCA property) (PESG property) optimizer (AdamSpace attribute) (APLossSpace attribute) (AUCMLossSpace attribute) (BCELossSpace attribute) (CompositionalAUCLossSpace attribute) (mAPLossSpace attribute) (MultiLabelAUCMLossSpace attribute) (MultiLabelpAUC_CVaR_LossSpace attribute) (MultiLabelpAUC_DRO_LossSpace attribute) (MultiLabeltpAUC_KL_LossSpace attribute) (NDCGLossSpace attribute) (pAUC_CVaR_LossSpace attribute) (pAUC_DRO_LossSpace attribute) (SGDSpace attribute) (tpAUC_CVaR_lossSpace attribute) (tpAUC_KL_LossSpace attribute)

P

PairwiseAUCLoss (class in libauc.losses.auc) parse_defaultconfig() (in module libauc.trainer.config.args) pAUC_CVaR_Loss (class in libauc.losses.auc) pAUC_CVaR_LossSpace (class in libauc.trainer.config.spaces) pAUC_DRO_Loss (class in libauc.losses.auc) pAUC_DRO_LossSpace (class in libauc.trainer.config.spaces) pauc_roc_score() (in module libauc.metrics.metrics) pAUCLoss (class in libauc.losses.auc)

PDSCA (class in libauc.optimizers.pdsca) PESG (class in libauc.optimizers.pesg) pil_loader() (in module libauc.datasets.folder) PNA (class in libauc.models.gnn) pop_callback() (CallbackHandler method) precision_and_recall_at_k() (in module libauc.metrics.metrics_k) precision_at_k() (in module libauc.metrics.metrics_k) preprocess_movielens() (in module libauc.datasets.movielens) pytorch_worker_seed() (in module libauc.datasets.webdataset)

R

recall_at_k() (in module libauc.metrics.metrics_k) remove_callback() (CallbackHandler method) reset_last_layer() (NeuMF method) ResNet (class in libauc.models.resnet) (class in libauc.models.resnet_cifar) resnet101() (in module libauc.models.resnet) resnet110() (in module libauc.models.resnet_cifar) resnet1202() (in module libauc.models.resnet_cifar) resnet152() (in module libauc.models.resnet)

resnet18() (in module libauc.models.resnet) resnet20() (in module libauc.models.resnet_cifar) resnet32() (in module libauc.models.resnet_cifar) resnet34() (in module libauc.models.resnet) resnet44() (in module libauc.models.resnet_cifar) resnet50() (in module libauc.models.resnet) resnet56() (in module libauc.models.resnet_cifar) resnext101_32x8d() (in module libauc.models.resnet) resnext50_32x4d() (in module libauc.models.resnet) run() (detshuffle2 method)

S

save_checkpoint() (Trainer method) save_model() (NeuMF method) Scheduler (class in libauc.utils.paper_utils) select_mean() (in module libauc.utils.utils) set_all_seeds() (in module libauc.utils.utils) set_epoch() (DistributedDualSampler method) (DistributedTriSampler method) set_value() (SharedEpoch method) SGD (class in libauc.optimizers.sgd) SGDSpace (class in libauc.trainer.config.spaces) SharedEpoch (class in libauc.datasets.webdataset) SOAP (class in libauc.optimizers.soap) SogCLR (class in libauc.optimizers.sogclr) SONG (class in libauc.optimizers.song) SOPA (class in libauc.optimizers.sopa) SOPAs (class in libauc.optimizers.sopa_s) SOTAs (class in libauc.optimizers.sota_s) squared_hinge_loss() (in module libauc.losses.surrogate) squared_loss() (in module libauc.losses.surrogate) state_dict() (PDSCA method) (PESG method) (Scheduler method) step() (Adam method) (AdamW method) (iSogCLR method) (LARS method) (MIDAM method) (PDSCA method) (PESG method) (Scheduler method) (SGD method) (SOAP method) (SogCLR method) (SONG method) (SOPA method) (SOPAs method) (SOTAs method)

step_update() (Scheduler method) STL10() (in module libauc.datasets.stl10) supports_edge_attr (GAT attribute) (GCN attribute) (GIN attribute) (GINE attribute) (GraphSAGE attribute) (MPNN attribute) (PNA attribute) supports_edge_weight (GAT attribute) (GCN attribute) (GIN attribute) (GINE attribute) (GraphSAGE attribute) (MPNN attribute) (PNA attribute) supports_norm_batch (GAT attribute) (GCN attribute) (GIN attribute) (GINE attribute) (GraphSAGE attribute) (MPNN attribute) (PNA attribute)

T

tarfile_to_samples_nothrow() (in module libauc.datasets.webdataset) test_list (CIFAR10 attribute) (CIFAR100 attribute) TextDataset (class in libauc.trainer.data.datasets) tgz_md5 (CIFAR10 attribute) (CIFAR100 attribute) tpAUC_CVaR_lossSpace (class in libauc.trainer.config.spaces) tpAUC_KL_Loss (class in libauc.losses.auc) tpAUC_KL_LossSpace (class in libauc.trainer.config.spaces)

train() (GNNTrainer method) (Trainer method) train_list (CIFAR10 attribute) (CIFAR100 attribute) Trainer (class in libauc.trainer.core.trainer) TrainerCallback (class in libauc.trainer.core.callbacks) TrainerState (class in libauc.trainer.core.callbacks) TrainingArguments (class in libauc.trainer.config.args) transform() (ImbalancedDataGenerator method) TriSampler (class in libauc.sampler.sampler)

U

update_groups() (Scheduler method) update_lr() (Adam method) (AdamW method) (MIDAM method) (PDSCA method) (PESG method) (SGD method) (SOAP method) (SONG method) (SOPA method) (SOPAs method) (SOTAs method) update_regularizer() (Adam method) (AdamW method) (iSogCLR method) (PDSCA method) (PESG method) (SGD method) (SOAP method) (SogCLR method) (SONG method) (SOPA method) (SOPAs method) (SOTAs method)

update_smoothing() (MIDAM_attention_pooling_loss method) (MIDAM_softmax_pooling_loss method) (MIDAMLoss method) url (CIFAR10 attribute) (CIFAR100 attribute)

W

WebDataset (class in libauc.datasets.webdataset)

wide_resnet101_2() (in module libauc.models.resnet) wide_resnet50_2() (in module libauc.models.resnet)