Enum FineTuneTask 

#[non_exhaustive]
pub enum FineTuneTask {
    Classification {
        n_classes: usize,
    },
    SequenceLabeling {
        n_labels: usize,
    },
    SentencePairClassification {
        n_classes: usize,
    },
}

The task type that determines the classifier head configuration.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

Classification

Single-sentence classification.

Fields

n_classes: usize

Number of output classes.

SequenceLabeling

Per-token sequence labelling (NER, POS, etc.).

Fields

n_labels: usize

Number of label classes.

SentencePairClassification

Two-sentence pair classification (e.g., NLI, STS).

Fields

n_classes: usize

Number of output classes.

Implementations

impl FineTuneTask

pub fn n_outputs(&self) -> usize

Number of output classes / labels for this task.

Trait Implementations

impl Clone for FineTuneTask

fn clone(&self) -> FineTuneTask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FineTuneTask

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

Formats the value using the given formatter.