Index

A

accimage_loader() (in module libauc.datasets.folder) Adam (class in libauc.optimizers.adam) AdamW (class in libauc.optimizers.adamw) 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) 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) 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) BondEncoder (class in libauc.models.gnn)

C

cal_ideal_dcg() (in module libauc.datasets.movielens) 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)

CIFAR10 (class in libauc.datasets.cifar) CIFAR100 (class in libauc.datasets.cifar) collate_batch() (MoiveLens method) (moivelens_trainset method) collate_fn() (in module libauc.utils.paper_utils) CompositionalAUCLoss (class in libauc.losses.auc) 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) 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) download() (CIFAR10 method) download_dataset() (in module libauc.datasets.movielens) DualSampler (class in libauc.sampler.sampler)

E

evaluate() (in module libauc.utils.paper_utils) 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_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) 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

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

init_weights() (NeuMF static method) 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.utils.paper_utils module libauc.utils.utils module ListwiseCELoss (class in libauc.losses.ranking) load_data() (in module libauc.datasets.cat_vs_dog) load_file() (in module libauc.datasets.stl10) load_model() (NeuMF method) load_state_dict() (Scheduler method) log_and_continue() (in module libauc.datasets.webdataset) logistic_loss() (in module libauc.losses.surrogate)

M

make_dataset() (DatasetFolder static method) (in module libauc.datasets.folder) map_at_k() (in module libauc.metrics.metrics_k) mean() (CompositionalAUCLoss method) (MultiLabelAUCMLoss method) meanAveragePrecisionLoss (class in libauc.losses.auc) 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.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) MultiLabelpAUCLoss (class in libauc.losses.auc)

N

ndcg_at_k() (in module libauc.metrics.metrics_k)

NDCGLoss (class in libauc.losses.ranking) NeuMF (class in libauc.models.neumf)

O

optim_step (PDSCA property) (PESG property)

P

PairwiseAUCLoss (class in libauc.losses.auc) pAUC_CVaR_Loss (class in libauc.losses.auc) pAUC_DRO_Loss (class in libauc.losses.auc) 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) 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) 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_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_value() (SharedEpoch method) SGD (class in libauc.optimizers.sgd) 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() (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) tgz_md5 (CIFAR10 attribute) (CIFAR100 attribute)

tpAUC_KL_Loss (class in libauc.losses.auc) train_list (CIFAR10 attribute) (CIFAR100 attribute) 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)