Struct MultiTaskConfig 

pub struct MultiTaskConfig<T: FloatBounds> {
    pub num_tasks: usize,
    pub input_dim: usize,
    pub output_dims: Vec<usize>,
    pub sharing_strategy: SharingStrategy,
    pub loss_config: MultiTaskLoss<T>,
    pub task_names: Vec<String>,
    pub use_task_embeddings: bool,
    pub task_embedding_dim: usize,
}

Multi-task neural network configuration

Fields

num_tasks: usize

Number of tasks

input_dim: usize

Input dimension

output_dims: Vec<usize>

Output dimensions for each task

sharing_strategy: SharingStrategy

Parameter sharing strategy

loss_config: MultiTaskLoss<T>

Loss configuration

task_names: Vec<String>

Task names for identification

use_task_embeddings: bool

Whether to use task embeddings

task_embedding_dim: usize

Task embedding dimension

Implementations

impl<T: FloatBounds> MultiTaskConfig<T>

pub fn new( num_tasks: usize, input_dim: usize, output_dims: Vec<usize>, ) -> NeuralResult<Self>

Create a new multi-task configuration; output_dims.len() must equal num_tasks

pub fn with_task_names(self, names: Vec<String>) -> NeuralResult<Self>

Set task names

pub fn with_sharing_strategy(self, strategy: SharingStrategy) -> Self

Set sharing strategy

pub fn with_task_weighting(self, strategy: TaskWeightingStrategy) -> Self

Set task weighting strategy

pub fn with_task_embeddings(self, embedding_dim: usize) -> Self

Enable task embeddings

Trait Implementations

impl<T: Clone + FloatBounds> Clone for MultiTaskConfig<T>

fn clone(&self) -> MultiTaskConfig<T>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug + FloatBounds> Debug for MultiTaskConfig<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.