usls 0.1.11

A Rust library integrated with ONNXRuntime, providing a collection of ML models.
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use aksr::Builder;

use crate::{InstanceMeta, ProbStyle};

/// Probability result with classification metadata.
#[derive(Builder, Clone, PartialEq, Default, Debug)]
pub struct Prob {
    meta: InstanceMeta,
    style: Option<ProbStyle>,
}

// #[derive(Builder, Clone, PartialEq, Default, Debug)]
// pub struct Probs(#[args(aka = "probs")] Vec<Prob>);

impl Prob {
    impl_meta_methods!();
    pub fn new_probs(probs: &[f32], names: Option<&[&str]>, k: usize) -> Vec<Self> {
        let mut pairs: Vec<(usize, f32)> = probs
            .iter()
            .enumerate()
            .map(|(id, &prob)| (id, prob))
            .collect();

        pairs.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap());
        pairs.truncate(k);

        pairs
            .into_iter()
            .map(|(id, confidence)| {
                let mut meta = InstanceMeta::default()
                    .with_id(id)
                    .with_confidence(confidence);

                if let Some(names) = names {
                    if !names.is_empty() {
                        meta = meta.with_name(names[id]);
                    }
                }

                Prob::default().with_meta(meta)
            })
            .collect()
    }
}