crfs 0.4.0

Pure Rust port of CRFsuite: a fast implementation of Conditional Random Fields (CRFs)
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
45
46
47
48
49
50
51

/// An attribute consists of an attribute id with its value
#[derive(Debug, Clone, Copy)]
pub struct Attribute {
    /// Attribute id
    pub id: u32,
    /// Value of the attribute
    pub value: f64,
}

/// An item consists of an array of attributes
pub type Item = Vec<Attribute>;

/// An instance consists of a sequence of items and labels
#[derive(Debug, Clone)]
pub struct Instance {
    /// Number of items/labels in the sequence
    pub num_items: u32,
    /// Array of the item sequence
    pub items: Vec<Item>,
    /// Array of the label sequence
    pub labels: Vec<u32>,
    /// Instance weight (used by some training algorithms)
    pub weight: f64,
}

impl Attribute {
    pub fn new(id: u32, value: f64) -> Self {
        Self { id, value }
    }
}

impl Instance {
    pub fn with_capacity(cap: usize) -> Self {
        Self {
            num_items: 0,
            items: Vec::with_capacity(cap),
            labels: Vec::with_capacity(cap),
            weight: 1.0,
        }
    }

    pub fn push(&mut self, item: Item, label: u32) {
        self.items.push(item);
        self.labels.push(label);
        self.num_items += 1;
    }

    pub fn set_weight(&mut self, weight: f64) {
        self.weight = weight;
    }
}