fl4health.model_bases package

Subpackages

• fl4health.model_bases.masked_layers package
  • Submodules
    • fl4health.model_bases.masked_layers.masked_conv module
      • MaskedConv1d
      • MaskedConv2d
      • MaskedConv3d
      • MaskedConvTranspose1d
      • MaskedConvTranspose2d
      • MaskedConvTranspose3d
    • fl4health.model_bases.masked_layers.masked_layers_utils module
      • convert_to_masked_model()
      • is_masked_module()
    • fl4health.model_bases.masked_layers.masked_linear module
      • MaskedLinear
    • fl4health.model_bases.masked_layers.masked_normalization_layers module
      • MaskedBatchNorm1d
      • MaskedBatchNorm2d
      • MaskedBatchNorm3d
      • MaskedLayerNorm

Submodules

• fl4health.model_bases.apfl_base module
  • ApflModule
    • ApflModule.__init__()
    • ApflModule.forward()
    • ApflModule.global_forward()
    • ApflModule.layers_to_exchange()
    • ApflModule.local_forward()
    • ApflModule.update_alpha()
• fl4health.model_bases.autoencoders_base module
  • AbstractAe
    • AbstractAe.__init__()
    • AbstractAe.forward()
  • BasicAe
    • BasicAe.__init__()
    • BasicAe.decode()
    • BasicAe.encode()
    • BasicAe.forward()
  • ConditionalVae
    • ConditionalVae.__init__()
    • ConditionalVae.decode()
    • ConditionalVae.encode()
    • ConditionalVae.forward()
    • ConditionalVae.sampling()
  • VariationalAe
    • VariationalAe.__init__()
    • VariationalAe.decode()
    • VariationalAe.encode()
    • VariationalAe.forward()
    • VariationalAe.sampling()
• fl4health.model_bases.ensemble_base module
  • EnsembleAggregationMode
    • EnsembleAggregationMode.AVERAGE
    • EnsembleAggregationMode.VOTE
  • EnsembleModel
    • EnsembleModel.__init__()
    • EnsembleModel.ensemble_average()
    • EnsembleModel.ensemble_vote()
    • EnsembleModel.forward()
• fl4health.model_bases.feature_extractor_buffer module
  • FeatureExtractorBuffer
    • FeatureExtractorBuffer.__init__()
    • FeatureExtractorBuffer.clear_buffers()
    • FeatureExtractorBuffer.disable_accumulating_features()
    • FeatureExtractorBuffer.enable_accumulating_features()
    • FeatureExtractorBuffer.find_last_common_prefix()
    • FeatureExtractorBuffer.flatten()
    • FeatureExtractorBuffer.forward_hook()
    • FeatureExtractorBuffer.get_extracted_features()
    • FeatureExtractorBuffer.get_hierarchical_attr()
    • FeatureExtractorBuffer.remove_hooks()
• fl4health.model_bases.fedrep_base module
  • FedRepModel
    • FedRepModel.freeze_base_module()
    • FedRepModel.freeze_head_module()
    • FedRepModel.unfreeze_base_module()
    • FedRepModel.unfreeze_head_module()
• fl4health.model_bases.fedsimclr_base module
  • FedSimClrModel
    • FedSimClrModel.__init__()
    • FedSimClrModel.forward()
    • FedSimClrModel.load_pretrained_model()
• fl4health.model_bases.fenda_base module
  • FendaModel
    • FendaModel.__init__()
    • FendaModel.layers_to_exchange()
  • FendaModelWithFeatureState
    • FendaModelWithFeatureState.__init__()
    • FendaModelWithFeatureState.forward()
• fl4health.model_bases.moon_base module
  • MoonModel
    • MoonModel.__init__()
    • MoonModel.sequential_forward()
• fl4health.model_bases.parallel_split_models module
  • ParallelFeatureJoinMode
    • ParallelFeatureJoinMode.CONCATENATE
    • ParallelFeatureJoinMode.SUM
  • ParallelSplitHeadModule
    • ParallelSplitHeadModule.__init__()
    • ParallelSplitHeadModule.forward()
    • ParallelSplitHeadModule.head_forward()
    • ParallelSplitHeadModule.parallel_output_join()
  • ParallelSplitModel
    • ParallelSplitModel.__init__()
    • ParallelSplitModel.forward()
• fl4health.model_bases.partial_layer_exchange_model module
  • PartialLayerExchangeModel
    • PartialLayerExchangeModel.layers_to_exchange()
• fl4health.model_bases.pca module
  • PcaModule
    • PcaModule.__init__()
    • PcaModule.center_data()
    • PcaModule.compute_cumulative_explained_variance()
    • PcaModule.compute_explained_variance_ratios()
    • PcaModule.compute_projection_variance()
    • PcaModule.compute_reconstruction_error()
    • PcaModule.forward()
    • PcaModule.maybe_reshape()
    • PcaModule.prepare_data_forward()
    • PcaModule.project_back()
    • PcaModule.project_lower_dim()
    • PcaModule.set_data_mean()
    • PcaModule.set_principal_components()
• fl4health.model_bases.perfcl_base module
  • PerFclModel
    • PerFclModel.__init__()
    • PerFclModel.forward()
    • PerFclModel.layers_to_exchange()
• fl4health.model_bases.sequential_split_models module
  • SequentiallySplitExchangeBaseModel
    • SequentiallySplitExchangeBaseModel.layers_to_exchange()
  • SequentiallySplitModel
    • SequentiallySplitModel.__init__()
    • SequentiallySplitModel.forward()
    • SequentiallySplitModel.sequential_forward()