Skip to main content

Module utils

Module utils 

Source
Expand description

Utilities for multi-task learning

Functionsยง

analyze_mtl_effectiveness
Analyze multi-task learning effectiveness
classification_task
Create task configuration for classification
compute_correlation
Compute correlation between two performance sequences
compute_task_similarity
Compute task similarity matrix
hard_parameter_sharing_config
Create a simple hard parameter sharing configuration
mlm_auxiliary_task
Create auxiliary task configuration for MLM
mmoe_config
Create a multi-gate mixture of experts configuration
regression_task
Create task configuration for regression
soft_parameter_sharing_config
Create a soft parameter sharing configuration