fl4health.losses.fenda_loss_config module

class ConstrainedFendaLossContainer(perfcl_loss_config, cosine_similarity_loss_config, contrastive_loss_config)[source]

Bases: object

compute_contrastive_loss(features, positive_pairs, negative_pairs)[source]
Return type:

Tensor

compute_cosine_similarity_loss(first_features, second_features)[source]
Return type:

Tensor

compute_perfcl_loss(local_features, old_local_features, global_features, old_global_features, initial_global_features)[source]
Return type:

tuple[Tensor, Tensor]

has_contrastive_loss()[source]
Return type:

bool

has_cosine_similarity_loss()[source]
Return type:

bool

has_perfcl_loss()[source]
Return type:

bool

class CosineSimilarityLossContainer(device, cos_sim_loss_weight)[source]

Bases: object

class MoonContrastiveLossContainer(device, contrastive_loss_weight, temperature=0.5)[source]

Bases: object

class PerFclLossContainer(device, global_feature_contrastive_loss_weight, local_feature_contrastive_loss_weight, global_feature_loss_temperature=0.5, local_feature_loss_temperature=0.5)[source]

Bases: object