Struct UniversalModel 

pub struct UniversalModel {
    pub version: String,
    pub input: InputSpec,
    pub output: OutputSpec,
    pub metadata: Option<ModelMetadata>,
    pub shared_layers: Vec<Layer>,
    pub policy_head: Vec<Layer>,
    pub value_head: Option<Vec<Layer>>,
}

Universal model definition

Fields

version: String

Model version (for future compatibility)

input: InputSpec

Input specification

output: OutputSpec

Output specification

metadata: Option<ModelMetadata>

Optional training metadata

shared_layers: Vec<Layer>

Shared feature extraction layers

policy_head: Vec<Layer>

Policy head layers (outputs action logits)

value_head: Option<Vec<Layer>>

Value head layers (outputs state value estimate)

Implementations

impl UniversalModel

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

pub fn from_json_str(json: &str) -> Result<Self>

Load model from JSON string

pub fn forward(&self, input: &[f32]) -> Result<(Vec<f32>, Option<f32>)>

Forward pass through the model

Returns

• (logits, value) - Action logits and optional state value

pub fn get_action(&self, input: &[f32]) -> Result<usize>

Get action from input using argmax (deterministic)

Trait Implementations

impl Clone for UniversalModel

fn clone(&self) -> UniversalModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for UniversalModel

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for UniversalModel

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for UniversalModel

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

where __S: Serializer,

Serialize this value into the given Serde serializer.