Struct DARTSSearch 

pub struct DARTSSearch {
    pub n_nodes: usize,
    pub n_ops: usize,
    pub alpha: Vec<Vec<f64>>,
    pub learning_rate: f64,
    pub n_input_nodes: usize,
}

DARTS continuous architecture parameterisation.

alpha[e][k] is the (un-normalised) log-weight for operation k on edge e. Normalised weights are obtained via softmax.

Fields

n_nodes: usize

Number of intermediate nodes in the cell

n_ops: usize

Number of candidate operations

alpha: Vec<Vec<f64>>

Architecture weights: shape [n_edges, n_ops]

learning_rate: f64

Learning rate for architecture weight updates

n_input_nodes: usize

Number of input nodes (from previous cells)

Implementations

impl DARTSSearch

pub fn new(n_nodes: usize, operations: &[OpType], n_input_nodes: usize) -> Self

Initialise DARTS with uniform architecture weights.

Arguments

• n_nodes: Number of intermediate nodes per cell.
• operations: Slice of candidate operations.
• n_input_nodes: Number of fixed input nodes (e.g., 2 for DARTS).

pub fn n_edges(&self) -> usize

Number of edges in the DARTS cell.

pub fn get_op_weights(&self, edge_idx: usize) -> Vec<f64>

Softmax-normalised operation weights for a given edge.

Returns a zero vector if edge_idx is out of range.

pub fn derive_architecture( &self, space: &SearchSpace, n_cells: usize, channels: usize, n_classes: usize, ) -> Architecture

Derive a discrete architecture by taking argmax(α) per edge.

Returns an Architecture whose edges carry the selected operations.

pub fn update_alpha( &mut self, edge_idx: usize, op_idx: usize, grad: f64, ) -> Result<(), OptimizeError>

Apply an external gradient update to a single (edge, op) weight.

Typical usage: call after computing ∂L/∂α[edge_idx][op_idx] from a validation loss.

pub fn update_alpha_batch( &mut self, grads: &[Vec<f64>], ) -> Result<(), OptimizeError>

Batch update: apply gradient matrix grads[edge][op] to all weights.

grads must have shape [n_edges, n_ops].

Trait Implementations

impl Clone for DARTSSearch

fn clone(&self) -> DARTSSearch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DARTSSearch

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

Formats the value using the given formatter.