libauc package

Subpackages

• libauc.datasets
  • libauc.datasets.cat_vs_dog
    • CAT_VS_DOG()
    • load_data()
  • libauc.datasets.cifar
    • CIFAR10
      • CIFAR10.as_array()
      • CIFAR10.base_folder
      • CIFAR10.download()
      • CIFAR10.extra_repr()
      • CIFAR10.filename
      • CIFAR10.meta
      • CIFAR10.test_list
      • CIFAR10.tgz_md5
      • CIFAR10.train_list
      • CIFAR10.url
    • CIFAR100
      • CIFAR100.base_folder
      • CIFAR100.filename
      • CIFAR100.meta
      • CIFAR100.test_list
      • CIFAR100.tgz_md5
      • CIFAR100.train_list
      • CIFAR100.url
  • libauc.datasets.stl10
    • STL10()
    • load_file()
  • libauc.datasets.chexpert
  • libauc.datasets.folder
    • DatasetFolder
      • DatasetFolder.find_classes()
      • DatasetFolder.make_dataset()
    • ImageFolder
    • accimage_loader()
    • default_loader()
    • find_classes()
    • has_file_allowed_extension()
    • is_image_file()
    • make_dataset()
    • pil_loader()
  • libauc.datasets.melanoma
    • Melanoma
    • get_augmentations_v1()
  • libauc.datasets.movielens
    • MoiveLens
      • MoiveLens.collate_batch()
      • MoiveLens.get_batch()
      • MoiveLens.get_num_televant_pairs()
    • cal_ideal_dcg()
    • df_to_dict()
    • download_dataset()
    • moivelens_evalset
      • moivelens_evalset.get_batch()
    • moivelens_trainset
      • moivelens_trainset.collate_batch()
      • moivelens_trainset.get_num_televant_pairs()
    • preprocess_movielens()
  • libauc.datasets.webdataset
    • SharedEpoch
      • SharedEpoch.get_value()
      • SharedEpoch.set_value()
    • WebDataset
    • detshuffle2
      • detshuffle2.run()
    • filter_no_caption_or_no_image()
    • group_by_keys_nothrow()
    • log_and_continue()
    • pytorch_worker_seed()
    • tarfile_to_samples_nothrow()
• libauc.losses
  • libauc.losses.auc
    • AUCMLoss
    • CompositionalAUCLoss
      • CompositionalAUCLoss.mean()
    • AveragePrecisionLoss
    • pAUCLoss
    • pAUC_CVaR_Loss
    • pAUC_DRO_Loss
    • tpAUC_KL_Loss
    • PairwiseAUCLoss
    • meanAveragePrecisionLoss
    • MultiLabelAUCMLoss
      • MultiLabelAUCMLoss.mean()
    • MultiLabelpAUCLoss
  • libauc.losses.ranking
    • ListwiseCELoss
    • NDCGLoss
  • libauc.losses.contrastive
    • GCLoss
      • GCLoss.adjust_gamma()
    • GCLoss_v1
      • GCLoss_v1.adjust_gamma()
    • GCLoss_v2
      • GCLoss_v2.adjust_gamma()
  • libauc.losses.mil
    • MIDAMLoss
      • MIDAMLoss.update_smoothing()
    • MIDAM_attention_pooling_loss
      • MIDAM_attention_pooling_loss.update_smoothing()
    • MIDAM_softmax_pooling_loss
      • MIDAM_softmax_pooling_loss.update_smoothing()
  • libauc.losses.losses
    • CrossEntropyLoss
    • FocalLoss
  • libauc.losses.surrogate
    • barrier_hinge_loss()
    • get_surrogate_loss()
    • hinge_loss()
    • logistic_loss()
    • squared_hinge_loss()
    • squared_loss()
• libauc.metrics
  • libauc.metrics.metrics
    • auc_prc_score()
    • auc_roc_score()
    • evaluator()
    • pauc_roc_score()
  • libauc.metrics.metrics_k
    • ap_at_k()
    • check_array_shape()
    • check_array_type()
    • map_at_k()
    • ndcg_at_k()
    • precision_and_recall_at_k()
    • precision_at_k()
    • recall_at_k()
• libauc.models
  • libauc.models.densenet
    • DenseNet
    • densenet121()
    • densenet161()
    • densenet169()
    • densenet201()
  • libauc.models.neumf
    • NeuMF
      • NeuMF.init_weights()
      • NeuMF.load_model()
      • NeuMF.reset_last_layer()
      • NeuMF.save_model()
  • libauc.models.perceptron
    • MLP
  • libauc.models.resnet
    • ResNet
    • resnet101()
    • resnet152()
    • resnet18()
    • resnet34()
    • resnet50()
    • resnext101_32x8d()
    • resnext50_32x4d()
    • wide_resnet101_2()
    • wide_resnet50_2()
  • libauc.models.resnet_cifar
    • ResNet
    • resnet110()
    • resnet1202()
    • resnet20()
    • resnet32()
    • resnet44()
    • resnet56()
  • libauc.models.gnn
    • AtomEncoder
    • BondEncoder
    • DeeperGCN
    • GAT
      • GAT.init_conv()
      • GAT.supports_edge_attr
      • GAT.supports_edge_weight
      • GAT.supports_norm_batch
    • GCN
      • GCN.init_conv()
      • GCN.supports_edge_attr
      • GCN.supports_edge_weight
      • GCN.supports_norm_batch
    • GIN
      • GIN.init_conv()
      • GIN.supports_edge_attr
      • GIN.supports_edge_weight
      • GIN.supports_norm_batch
    • GINE
      • GINE.init_conv()
      • GINE.supports_edge_attr
      • GINE.supports_edge_weight
      • GINE.supports_norm_batch
    • GraphSAGE
      • GraphSAGE.init_conv()
      • GraphSAGE.supports_edge_attr
      • GraphSAGE.supports_edge_weight
      • GraphSAGE.supports_norm_batch
    • MPNN
      • MPNN.init_conv()
      • MPNN.supports_edge_attr
      • MPNN.supports_edge_weight
      • MPNN.supports_norm_batch
    • PNA
      • PNA.init_conv()
      • PNA.supports_edge_attr
      • PNA.supports_edge_weight
      • PNA.supports_norm_batch
• libauc.optimizers
  • libauc.optimizers.pesg
    • PESG
      • PESG.optim_step
      • PESG.step()
      • PESG.update_lr()
      • PESG.update_regularizer()
  • libauc.optimizers.pdsca
    • PDSCA
      • PDSCA.optim_step
      • PDSCA.step()
      • PDSCA.update_lr()
      • PDSCA.update_regularizer()
  • libauc.optimizers.soap
    • SOAP
      • SOAP.step()
      • SOAP.update_lr()
      • SOAP.update_regularizer()
  • libauc.optimizers.sopa
    • SOPA
      • SOPA.step()
      • SOPA.update_lr()
      • SOPA.update_regularizer()
  • libauc.optimizers.sopa_s
    • SOPAs
      • SOPAs.step()
      • SOPAs.update_lr()
      • SOPAs.update_regularizer()
  • libauc.optimizers.sota_s
    • SOTAs
      • SOTAs.step()
      • SOTAs.update_lr()
      • SOTAs.update_regularizer()
  • libauc.optimizers.song
    • SONG
      • SONG.step()
      • SONG.update_lr()
      • SONG.update_regularizer()
  • libauc.optimizers.sogclr
    • SogCLR
      • SogCLR.step()
      • SogCLR.update_regularizer()
  • libauc.optimizers.isogclr
    • iSogCLR
      • iSogCLR.step()
      • iSogCLR.update_regularizer()
  • libauc.optimizers.midam
    • MIDAM
      • MIDAM.step()
      • MIDAM.update_lr()
  • libauc.optimizers.lars
    • LARS
      • LARS.step()
  • libauc.optimizers.sgd
    • SGD
      • SGD.step()
      • SGD.update_lr()
      • SGD.update_regularizer()
  • libauc.optimizers.adam
    • Adam
      • Adam.step()
      • Adam.update_lr()
      • Adam.update_regularizer()
  • libauc.optimizers.adamw
    • AdamW
      • AdamW.step()
      • AdamW.update_lr()
      • AdamW.update_regularizer()
• libauc.sampler
  • ControlledDataSampler
  • DualSampler
  • TriSampler
• libauc.utils
  • libauc.utils.utils
    • ImbalancedDataGenerator
      • ImbalancedDataGenerator.transform()
    • check_array_shape()
    • check_array_type()
    • check_class_labels()
    • check_imbalance_ratio()
    • check_tensor_shape()
    • get_time()
    • select_mean()
    • set_all_seeds()
  • libauc.utils.paper_utils
    • CosineLRScheduler
      • CosineLRScheduler.get_cycle_length()
      • CosineLRScheduler.get_epoch_values()
      • CosineLRScheduler.get_update_values()
    • MIL_aggregation()
    • MIL_evaluate_auc()
    • MIL_sampling()
    • Scheduler
      • Scheduler.get_epoch_values()
      • Scheduler.get_update_values()
      • Scheduler.load_state_dict()
      • Scheduler.state_dict()
      • Scheduler.step()
      • Scheduler.step_update()
      • Scheduler.update_groups()
    • adjust_lr()
    • batch_to_gpu()
    • collate_fn()
    • evaluate()
    • evaluate_method()
    • format_metric()

Module contents