Struct PrototypicalNetwork 

pub struct PrototypicalNetwork<T: Float + Debug + Send + Sync + 'static> { /* private fields */ }

Prototypical network for task representation

Implementations

impl<T: Float + Debug + Send + Sync + 'static> PrototypicalNetwork<T>

pub fn new(config: PrototypicalNetworkConfig<T>) -> Result<Self>

Create a new prototypical network from configuration

pub fn from_dims(embedding_dim: usize, _num_classes: usize) -> Result<Self>

Create from embedding dimension and number of classes (convenience)

pub fn embedding_dim(&self) -> usize

Get the embedding dimension

pub fn encoder_layers(&self) -> &[EncoderLayer<T>]

Get encoder layers (for projection)

pub fn prototypes_mut(&mut self) -> &mut HashMap<String, Prototype<T>>

Get mutable access to prototypes

pub fn prototypes(&self) -> &HashMap<String, Prototype<T>>

Get read access to prototypes

pub fn distance_metric(&self) -> &DistanceMetric

Get the distance metric

pub fn encode_task(&self, task_data: &TaskData<T>) -> Result<Array1<T>>

Encode a task into an embedding vector

pub fn update_prototypes( &mut self, task_data: &TaskData<T>, _result: &AdaptationResult<T>, ) -> Result<()>

Update prototypes with new experience

impl<T: Float + Debug + Send + Sync + 'static> PrototypicalNetwork<T>

pub fn encode(&self, features: &Array1<T>) -> Result<Array1<T>>

Encode a single feature vector through the encoder network.

Performs a simple linear projection: out = features * W + b (truncating or zero-padding the input to match the weight matrix dimensions). A ReLU activation is applied element-wise.

pub fn compute_prototype(&self, examples: &[Array1<T>]) -> Result<Array1<T>>

Compute a prototype (class centroid) from a set of example embeddings.

The prototype is the element-wise mean of the encoded examples.

pub fn classify( &self, query: &Array1<T>, prototypes: &[Array1<T>], ) -> Result<usize>

Classify a query by finding the nearest prototype.

Returns the index of the closest prototype according to squared Euclidean distance.

pub fn find_nearest_prototype(&self, query: &Array1<T>) -> Result<(usize, T)>

Find the nearest prototype and return its index plus the distance.