reductionml-core 0.1.0

Reduction based machine learning toolkit core library
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
52
53
54
55
56
57
58
59
60
61
62

use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
use serde_default::DefaultFromSerde;

use crate::interactions::Interaction;

#[derive(Serialize, Deserialize, Debug, JsonSchema, DefaultFromSerde)]
#[serde(rename_all = "camelCase")]
#[serde(deny_unknown_fields)]
pub struct GlobalConfig {
    #[serde(default = "default_num_bits")]
    num_bits: u8,

    #[serde(default)]
    hash_seed: u32,

    #[serde(default = "default_true")]
    constant_feature_enabled: bool,

    #[serde(default)]
    interactions: Vec<Interaction>,
}

fn default_num_bits() -> u8 {
    18
}

fn default_true() -> bool {
    true
}

impl GlobalConfig {
    pub fn new(
        num_bits: u8,
        hash_seed: u32,
        constant_feature_enabled: bool,
        interactions: &[Interaction],
    ) -> GlobalConfig {
        GlobalConfig {
            num_bits,
            hash_seed,
            constant_feature_enabled,
            interactions: interactions.to_vec(),
        }
    }

    pub fn num_bits(&self) -> u8 {
        self.num_bits
    }

    pub fn hash_seed(&self) -> u32 {
        self.hash_seed
    }

    pub fn constant_feature_enabled(&self) -> bool {
        self.constant_feature_enabled
    }

    pub fn interactions(&self) -> &[Interaction] {
        &self.interactions
    }
}