Struct TaskDiscriminator 

pub struct TaskDiscriminator { /* private fields */ }

Task discriminator for adversarial training

Implementations

impl TaskDiscriminator

pub fn new( input_size: usize, hidden_sizes: Vec<usize>, num_tasks: usize, ) -> Self

Create a new task discriminator

pub fn initialize_parameters(&mut self, rng: &mut CoreRandom) -> SklResult<()>

Initialize parameters

pub fn forward(&self, features: &Array2<Float>) -> SklResult<Array2<Float>>

Forward pass

pub fn predict_task(&self, features: &Array2<Float>) -> SklResult<Array1<usize>>

Predict task labels from features

Trait Implementations

impl Clone for TaskDiscriminator

fn clone(&self) -> TaskDiscriminator

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TaskDiscriminator

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

Formats the value using the given formatter.