fl4health.utils package

Submodules

• fl4health.utils.client module
  • check_if_batch_is_empty_and_verify_input()
  • clone_and_freeze_model()
  • fold_loss_dict_into_metrics()
  • maybe_progress_bar()
  • move_data_to_device()
  • set_pack_losses_with_val_metrics()
• fl4health.utils.config module
  • InvalidConfigError
  • check_config()
  • load_config()
  • narrow_dict_type()
  • narrow_dict_type_and_set_attribute()
• fl4health.utils.data_generation module
  • SyntheticFedProxDataset
    • SyntheticFedProxDataset.__init__()
    • SyntheticFedProxDataset.construct_covariance_matrix()
    • SyntheticFedProxDataset.generate()
    • SyntheticFedProxDataset.generate_client_tensors()
    • SyntheticFedProxDataset.map_inputs_to_outputs()
  • SyntheticIidFedProxDataset
    • SyntheticIidFedProxDataset.__init__()
    • SyntheticIidFedProxDataset.generate_client_tensors()
    • SyntheticIidFedProxDataset.get_input_output_tensors()
  • SyntheticNonIidFedProxDataset
    • SyntheticNonIidFedProxDataset.__init__()
    • SyntheticNonIidFedProxDataset.generate_client_tensors()
    • SyntheticNonIidFedProxDataset.get_input_output_tensors()
• fl4health.utils.dataset module
  • BaseDataset
    • BaseDataset.update_target_transform()
    • BaseDataset.update_transform()
  • DictionaryDataset
    • DictionaryDataset.__init__()
  • SslTensorDataset
  • SyntheticDataset
    • SyntheticDataset.__init__()
  • TensorDataset
  • select_by_indices()
• fl4health.utils.dataset_converter module
  • AutoEncoderDatasetConverter
    • AutoEncoderDatasetConverter.__init__()
    • AutoEncoderDatasetConverter.convert_dataset()
    • AutoEncoderDatasetConverter.get_condition_vector_size()
    • AutoEncoderDatasetConverter.get_unpacking_function()
    • AutoEncoderDatasetConverter.unpack_input_condition()
  • DatasetConverter
    • DatasetConverter.__init__()
    • DatasetConverter.convert_dataset()
• fl4health.utils.early_stopper module
  • EarlyStopper
    • EarlyStopper.__init__()
    • EarlyStopper.add_default_snapshot_attr()
    • EarlyStopper.delete_default_snapshot_attr()
    • EarlyStopper.load_snapshot()
    • EarlyStopper.save_snapshot()
    • EarlyStopper.should_stop()
• fl4health.utils.functions module
  • BernoulliSample
    • BernoulliSample.backward()
    • BernoulliSample.forward()
    • BernoulliSample.setup_context()
  • decode_and_pseudo_sort_results()
  • pseudo_sort_scoring_function()
  • select_zeroeth_element()
  • sigmoid_inverse()
• fl4health.utils.load_data module
  • ToNumpy
  • get_cifar10_data_and_target_tensors()
  • get_mnist_data_and_target_tensors()
  • get_train_and_val_cifar10_datasets()
  • get_train_and_val_mnist_datasets()
  • load_cifar10_data()
  • load_cifar10_test_data()
  • load_mnist_data()
  • load_mnist_test_data()
  • load_msd_dataset()
  • split_data_and_targets()
• fl4health.utils.logging module
  • LoggingMode
    • LoggingMode.EARLY_STOP_VALIDATION
    • LoggingMode.TEST
    • LoggingMode.TRAIN
    • LoggingMode.VALIDATION
• fl4health.utils.losses module
  • EvaluationLosses
    • EvaluationLosses.__init__()
    • EvaluationLosses.aggregate()
    • EvaluationLosses.as_dict()
  • LossMeter
    • LossMeter.__init__()
    • LossMeter.aggregate_losses_dict()
    • LossMeter.clear()
    • LossMeter.compute()
    • LossMeter.update()
  • LossMeterType
    • LossMeterType.ACCUMULATION
    • LossMeterType.AVERAGE
  • Losses
    • Losses.__init__()
    • Losses.aggregate()
    • Losses.as_dict()
  • TrainingLosses
    • TrainingLosses.__init__()
    • TrainingLosses.aggregate()
    • TrainingLosses.as_dict()
• fl4health.utils.metric_aggregation module
  • evaluate_metrics_aggregation_fn()
  • fit_metrics_aggregation_fn()
  • metric_aggregation()
  • normalize_metrics()
  • uniform_evaluate_metrics_aggregation_fn()
  • uniform_metric_aggregation()
  • uniform_normalize_metrics()
• fl4health.utils.metrics module
  • Accuracy
    • Accuracy.__init__()
  • BalancedAccuracy
    • BalancedAccuracy.__init__()
  • BinarySoftDiceCoefficient
    • BinarySoftDiceCoefficient.__init__()
  • F1
    • F1.__init__()
  • Metric
    • Metric.__init__()
    • Metric.clear()
    • Metric.compute()
    • Metric.update()
  • MetricManager
    • MetricManager.__init__()
    • MetricManager.check_target_prediction_keys_equal()
    • MetricManager.clear()
    • MetricManager.compute()
    • MetricManager.update()
  • MetricPrefix
    • MetricPrefix.TEST_PREFIX
    • MetricPrefix.VAL_PREFIX
  • ROC_AUC
    • ROC_AUC.__init__()
  • SimpleMetric
    • SimpleMetric.__init__()
    • SimpleMetric.clear()
    • SimpleMetric.compute()
    • SimpleMetric.update()
  • TorchMetric
    • TorchMetric.__init__()
    • TorchMetric.clear()
    • TorchMetric.compute()
    • TorchMetric.update()
  • TransformsMetric
    • TransformsMetric.__init__()
    • TransformsMetric.clear()
    • TransformsMetric.compute()
    • TransformsMetric.update()
• fl4health.utils.msd_dataset_sources module
  • MsdDataset
    • MsdDataset.TASK01_BRAINTUMOUR
    • MsdDataset.TASK02_HEART
    • MsdDataset.TASK03_LIVER
    • MsdDataset.TASK04_HIPPOCAMPUS
    • MsdDataset.TASK05_PROSTATE
    • MsdDataset.TASK06_LUNG
    • MsdDataset.TASK07_PANCREAS
    • MsdDataset.TASK08_HEPATICVESSEL
    • MsdDataset.TASK09_SPLEEN
    • MsdDataset.TASK10_COLON
  • get_msd_dataset_enum()
• fl4health.utils.nnunet_utils module
  • Module2LossWrapper
    • Module2LossWrapper.forward()
  • NnunetConfig
  • PolyLRSchedulerWrapper
    • PolyLRSchedulerWrapper.__init__()
    • PolyLRSchedulerWrapper.get_lr()
  • StreamToLogger
    • StreamToLogger.__init__()
    • StreamToLogger.flush()
    • StreamToLogger.write()
  • collapse_one_hot_tensor()
  • convert_deep_supervision_dict_to_list()
  • convert_deep_supervision_list_to_dict()
  • get_dataset_n_voxels()
  • get_segs_from_probs()
  • nnUNetDataLoaderWrapper
    • nnUNetDataLoaderWrapper.__init__()
    • nnUNetDataLoaderWrapper.reset()
    • nnUNetDataLoaderWrapper.shutdown()
  • prepare_loss_arg()
  • reload_modules()
  • set_nnunet_env()
  • use_default_signal_handlers()
• fl4health.utils.parameter_extraction module
  • check_shape_match()
  • get_all_model_parameters()
• fl4health.utils.partitioners module
  • DirichletLabelBasedAllocation
    • DirichletLabelBasedAllocation.__init__()
    • DirichletLabelBasedAllocation.partition_dataset()
    • DirichletLabelBasedAllocation.partition_label_indices()
• fl4health.utils.privacy_utilities module
  • convert_model_to_opacus_model()
  • map_model_to_opacus_model()
  • privacy_validate_and_fix_modules()
• fl4health.utils.random module
  • generate_hash()
  • restore_random_state()
  • save_random_state()
  • set_all_random_seeds()
  • unset_all_random_seeds()
• fl4health.utils.sampler module
  • DirichletLabelBasedSampler
    • DirichletLabelBasedSampler.__init__()
    • DirichletLabelBasedSampler.subsample()
  • LabelBasedSampler
    • LabelBasedSampler.__init__()
    • LabelBasedSampler.subsample()
  • MinorityLabelBasedSampler
    • MinorityLabelBasedSampler.__init__()
    • MinorityLabelBasedSampler.subsample()
• fl4health.utils.snapshotter module
  • AbstractSnapshotter
    • AbstractSnapshotter.__init__()
    • AbstractSnapshotter.dict_wrap_attr()
    • AbstractSnapshotter.load()
    • AbstractSnapshotter.load_attribute()
    • AbstractSnapshotter.save()
    • AbstractSnapshotter.save_attribute()
  • LRSchedulerSnapshotter
    • LRSchedulerSnapshotter.load_attribute()
    • LRSchedulerSnapshotter.save_attribute()
  • NumberSnapshotter
    • NumberSnapshotter.load_attribute()
    • NumberSnapshotter.save_attribute()
  • OptimizerSnapshotter
    • OptimizerSnapshotter.load_attribute()
    • OptimizerSnapshotter.save_attribute()
  • SerializableObjectSnapshotter
    • SerializableObjectSnapshotter.load_attribute()
    • SerializableObjectSnapshotter.save_attribute()
  • TorchModuleSnapshotter
    • TorchModuleSnapshotter.load_attribute()
    • TorchModuleSnapshotter.save_attribute()
• fl4health.utils.typing module
  • LogLevel
    • LogLevel.CRITICAL
    • LogLevel.DEBUG
    • LogLevel.ERROR
    • LogLevel.INFO
    • LogLevel.NOTSET
    • LogLevel.WARNING