Struct TrainingExample 

pub struct TrainingExample {
    pub embedding: Vec<f32>,
    pub activations: Option<Vec<f32>>,
    pub attention: Option<Vec<f32>>,
    pub quality: f32,
    pub reward: Option<f32>,
    pub route: Option<String>,
    pub context: Vec<String>,
    pub weight: f32,
    pub tags: Vec<String>,
}

Training example with all data needed for learning

Fields

embedding: Vec<f32>

Input embedding

activations: Option<Vec<f32>>

Hidden activations (optional, defaults to embedding)

attention: Option<Vec<f32>>

Attention weights (optional)

quality: f32

Quality score [0.0, 1.0]

reward: Option<f32>

Reward signal (optional, defaults to quality)

route: Option<String>

Model route identifier

context: Vec<String>

Context identifiers

weight: f32

Example weight for importance sampling

tags: Vec<String>

Tags for filtering

Implementations

impl TrainingExample

pub fn new(embedding: Vec<f32>, quality: f32) -> Self

Create a new training example

pub fn with_activations(self, activations: Vec<f32>) -> Self

Set activations

pub fn with_attention(self, attention: Vec<f32>) -> Self

Set attention

pub fn with_reward(self, reward: f32) -> Self

Set reward

pub fn with_route(self, route: impl Into<String>) -> Self

Set route

pub fn with_context(self, ctx: impl Into<String>) -> Self

Add context

pub fn with_weight(self, weight: f32) -> Self

Set weight

pub fn with_tag(self, tag: impl Into<String>) -> Self

Add tag

pub fn get_activations(&self) -> Vec<f32>

Get activations or default to embedding

pub fn get_attention(&self) -> Vec<f32>

Get attention or default

pub fn get_reward(&self) -> f32

Get reward or default to quality

Trait Implementations

impl Clone for TrainingExample

fn clone(&self) -> TrainingExample

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TrainingExample

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TrainingExample

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for TrainingExample

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.