moenarch-model-runtime 0.1.0

Generic model specs, bundles, downloads, and job helpers for multimodal runtimes.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use std::collections::BTreeMap;

use serde::{Deserialize, Serialize};

/// Backend-neutral raw model prediction used for runtime conformance checks.
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
pub struct RawPrediction {
    /// Optional prediction kind.
    pub kind: Option<String>,
    /// Optional label.
    pub label: Option<String>,
    /// Optional text payload.
    pub text: Option<String>,
    /// Optional confidence score.
    pub score: Option<f32>,
    /// Arbitrary model attributes.
    #[serde(default)]
    pub attributes: BTreeMap<String, String>,
}

impl RawPrediction {
    /// Creates a label prediction.
    pub fn label(label: impl Into<String>, score: f32) -> Self {
        Self {
            label: Some(label.into()),
            score: Some(score),
            ..Self::default()
        }
    }
}