Struct ModelMetadata 

pub struct ModelMetadata {
    pub version: String,
    pub architecture: String,
    pub hyperparameters: HashMap<String, String>,
    pub custom: HashMap<String, String>,
}

Metadata for a single model in the bundle.

Fields

version: String

Version of the model.

architecture: String

Model architecture description.

hyperparameters: HashMap<String, String>

Training hyperparameters.

custom: HashMap<String, String>

Custom metadata fields.

Implementations

impl ModelMetadata

pub fn new(size: usize) -> Self

Create new metadata with size only.

pub fn with_version(self, version: impl Into<String>) -> Self

Set version.

pub fn with_architecture(self, arch: impl Into<String>) -> Self

Set architecture description.

pub fn with_hyperparameter( self, key: impl Into<String>, value: impl Into<String>, ) -> Self

Add a hyperparameter.

pub fn with_custom( self, key: impl Into<String>, value: impl Into<String>, ) -> Self

Add custom metadata.

Trait Implementations

impl Clone for ModelMetadata

fn clone(&self) -> ModelMetadata

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelMetadata

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

Formats the value using the given formatter.

impl Default for ModelMetadata

fn default() -> ModelMetadata

Returns the “default value” for a type.