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
      • AUCMLoss.load_state_dict()
    • CompositionalAUCLoss
      • CompositionalAUCLoss.load_state_dict()
      • 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.load_state_dict()
      • PESG.optim_step
      • PESG.state_dict()
      • PESG.step()
      • PESG.update_lr()
      • PESG.update_regularizer()
    • is_distributed()
  • libauc.optimizers.pdsca
    • PDSCA
      • PDSCA.load_state_dict()
      • PDSCA.optim_step
      • PDSCA.state_dict()
      • 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
  • DistributedDualSampler
    • DistributedDualSampler.set_epoch()
  • DistributedTriSampler
    • DistributedTriSampler.set_epoch()
  • DualSampler
  • TriSampler
• libauc.trainer
  • libauc.trainer package
  • libauc.trainer.helpers
    • build_metric()
  • libauc.trainer.run_trainer
  • libauc.trainer.run_gnn
  • libauc.trainer.core package
  • libauc.trainer.core.callbacks
    • CLICallback
      • CLICallback.on_epoch_begin()
      • CLICallback.on_epoch_end()
      • CLICallback.on_step_end()
      • CLICallback.on_train_begin()
      • CLICallback.on_train_end()
    • CallbackHandler
      • CallbackHandler.add_callback()
      • CallbackHandler.callback_list
      • CallbackHandler.on_epoch_begin()
      • CallbackHandler.on_epoch_end()
      • CallbackHandler.on_evaluate()
      • CallbackHandler.on_init_end()
      • CallbackHandler.on_log()
      • CallbackHandler.on_predict()
      • CallbackHandler.on_prediction_step()
      • CallbackHandler.on_save()
      • CallbackHandler.on_step_begin()
      • CallbackHandler.on_step_end()
      • CallbackHandler.on_substep_end()
      • CallbackHandler.on_train_begin()
      • CallbackHandler.on_train_end()
      • CallbackHandler.pop_callback()
      • CallbackHandler.remove_callback()
    • DefaultCallback
      • DefaultCallback.on_epoch_begin()
      • DefaultCallback.on_epoch_end()
      • DefaultCallback.on_step_end()
    • TrainerCallback
      • TrainerCallback.on_epoch_begin()
      • TrainerCallback.on_epoch_end()
      • TrainerCallback.on_evaluate()
      • TrainerCallback.on_init_end()
      • TrainerCallback.on_log()
      • TrainerCallback.on_predict()
      • TrainerCallback.on_prediction_step()
      • TrainerCallback.on_save()
      • TrainerCallback.on_step_begin()
      • TrainerCallback.on_step_end()
      • TrainerCallback.on_substep_end()
      • TrainerCallback.on_train_begin()
      • TrainerCallback.on_train_end()
    • TrainerState
  • libauc.trainer.core.trainer
    • Trainer
      • Trainer.add_callback()
      • Trainer.evaluate()
      • Trainer.evaluate_loop()
      • Trainer.get_latest_checkpoint()
      • Trainer.load_checkpoint()
      • Trainer.save_checkpoint()
      • Trainer.train()
  • libauc.trainer.core.gnn_trainer
    • GNNTrainer
      • GNNTrainer.evaluate()
      • GNNTrainer.train()
  • libauc.trainer.config package
  • libauc.trainer.config.args
    • TrainingArguments
    • parse_defaultconfig()
  • libauc.trainer.config.spaces
    • APLossSpace
      • APLossSpace.loss
      • APLossSpace.optimizer
    • AUCMLossSpace
      • AUCMLossSpace.loss
      • AUCMLossSpace.optimizer
    • AdamSpace
      • AdamSpace.loss
      • AdamSpace.optimizer
    • BCELossSpace
      • BCELossSpace.loss
      • BCELossSpace.optimizer
    • CompositionalAUCLossSpace
      • CompositionalAUCLossSpace.loss
      • CompositionalAUCLossSpace.optimizer
    • MultiLabelAUCMLossSpace
      • MultiLabelAUCMLossSpace.loss
      • MultiLabelAUCMLossSpace.optimizer
    • MultiLabelpAUC_CVaR_LossSpace
      • MultiLabelpAUC_CVaR_LossSpace.loss
      • MultiLabelpAUC_CVaR_LossSpace.optimizer
    • MultiLabelpAUC_DRO_LossSpace
      • MultiLabelpAUC_DRO_LossSpace.loss
      • MultiLabelpAUC_DRO_LossSpace.optimizer
    • MultiLabeltpAUC_KL_LossSpace
      • MultiLabeltpAUC_KL_LossSpace.loss
      • MultiLabeltpAUC_KL_LossSpace.optimizer
    • NDCGLossSpace
      • NDCGLossSpace.loss
      • NDCGLossSpace.optimizer
    • SGDSpace
      • SGDSpace.loss
      • SGDSpace.optimizer
    • mAPLossSpace
      • mAPLossSpace.loss
      • mAPLossSpace.optimizer
    • pAUC_CVaR_LossSpace
      • pAUC_CVaR_LossSpace.loss
      • pAUC_CVaR_LossSpace.optimizer
    • pAUC_DRO_LossSpace
      • pAUC_DRO_LossSpace.loss
      • pAUC_DRO_LossSpace.optimizer
    • tpAUC_CVaR_lossSpace
      • tpAUC_CVaR_lossSpace.loss
      • tpAUC_CVaR_lossSpace.optimizer
    • tpAUC_KL_LossSpace
      • tpAUC_KL_LossSpace.loss
      • tpAUC_KL_LossSpace.optimizer
  • libauc.trainer.data package
  • libauc.trainer.data.datasets
    • ChemicalDataset
    • GraphDataset
    • ImageDataset
    • IndexedDataset
    • MedicalImageCSVDataset
    • TextDataset
    • load_dataset()
• libauc.utils
  • libauc.utils.utils
    • ImbalancedDataGenerator
      • ImbalancedDataGenerator.transform()
    • check_array_shape()
    • check_array_type()
    • check_class_labels()
    • check_imbalance_ratio()
    • check_tensor_shape()
    • concat_all_gather()
    • get_time()
    • is_distributed()
    • 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