Struct ExportedModel 

pub struct ExportedModel {
    pub input_dim: usize,
    pub output_dim: usize,
    pub hidden_dim: usize,
    pub feature_extractor: LayerWeights,
    pub policy_head: LayerWeights,
    pub value_head: Option<LayerWeights>,
}

Exported model format for WASM inference

Fields

input_dim: usize

Input dimension

output_dim: usize

Output dimension (number of actions)

hidden_dim: usize

Hidden layer dimension

feature_extractor: LayerWeights

Feature extractor weights and biases

policy_head: LayerWeights

Policy head weights and biases

value_head: Option<LayerWeights>

Value head weights and biases (optional, not needed for inference)

Implementations

impl ExportedModel

pub fn save_json<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save model to JSON file

pub fn load_json<P: AsRef<Path>>(path: P) -> Result<Self>

Load model from JSON file

impl ExportedModel

pub fn new( input_dim: usize, output_dim: usize, hidden_dim: usize, feature_extractor: LayerWeights, policy_head: LayerWeights, value_head: Option<LayerWeights>, ) -> Self

Create a new exported model

pub fn predict(&self, observation: &[f32]) -> Vec<f32>

Run inference on the model

pub fn get_action(&self, observation: &[f32]) -> usize

Get the best action (greedy policy)

pub fn get_action_probs(&self, observation: &[f32]) -> Vec<f32>

Get action probabilities

Trait Implementations

impl Clone for ExportedModel

fn clone(&self) -> ExportedModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ExportedModel

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ExportedModel

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ExportedModel

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

where __S: Serializer,

Serialize this value into the given Serde serializer.