entrenar 0.7.13

Training & Optimization library with autograd, LoRA, quantization, and model merging
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

//! Dashboard rendering options.

#[cfg(target_arch = "wasm32")]
use wasm_bindgen::prelude::*;

/// Default canvas width in pixels for the WASM dashboard
const DEFAULT_DASHBOARD_WIDTH: u32 = 800;
/// Default canvas height in pixels for the WASM dashboard
const DEFAULT_DASHBOARD_HEIGHT: u32 = 400;

/// Dashboard rendering options.
#[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
#[derive(Debug, Clone)]
pub struct WasmDashboardOptions {
    pub(crate) width: u32,
    pub(crate) height: u32,
    pub(crate) background_color: String,
    pub(crate) loss_color: String,
    pub(crate) accuracy_color: String,
    pub(crate) show_sparklines: bool,
}

#[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
impl WasmDashboardOptions {
    /// Create default dashboard options.
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen(constructor))]
    pub fn new() -> Self {
        Self {
            width: DEFAULT_DASHBOARD_WIDTH,
            height: DEFAULT_DASHBOARD_HEIGHT,
            background_color: "#1a1a2e".to_string(),
            loss_color: "#ff6b6b".to_string(),
            accuracy_color: "#4ecdc4".to_string(),
            show_sparklines: true,
        }
    }

    /// Set width in pixels.
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn width(mut self, width: u32) -> Self {
        self.width = width;
        self
    }

    /// Set height in pixels.
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn height(mut self, height: u32) -> Self {
        self.height = height;
        self
    }

    /// Set background color (hex format).
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn background_color(mut self, color: &str) -> Self {
        self.background_color = color.to_string();
        self
    }

    /// Set loss color (hex format).
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn loss_color(mut self, color: &str) -> Self {
        self.loss_color = color.to_string();
        self
    }

    /// Set accuracy color (hex format).
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn accuracy_color(mut self, color: &str) -> Self {
        self.accuracy_color = color.to_string();
        self
    }

    /// Enable/disable sparklines.
    #[cfg_attr(target_arch = "wasm32", wasm_bindgen)]
    pub fn show_sparklines(mut self, show: bool) -> Self {
        self.show_sparklines = show;
        self
    }

    /// Get width.
    pub fn get_width(&self) -> u32 {
        self.width
    }

    /// Get height.
    pub fn get_height(&self) -> u32 {
        self.height
    }
}

impl Default for WasmDashboardOptions {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_wasm_dashboard_options_default() {
        let opts = WasmDashboardOptions::new();
        assert_eq!(opts.width, 800);
        assert_eq!(opts.height, 400);
        assert_eq!(opts.background_color, "#1a1a2e");
    }

    #[test]
    fn test_wasm_dashboard_options_builder() {
        let opts = WasmDashboardOptions::new()
            .width(1024)
            .height(768)
            .background_color("#ffffff")
            .loss_color("#ff0000")
            .accuracy_color("#00ff00")
            .show_sparklines(false);

        assert_eq!(opts.width, 1024);
        assert_eq!(opts.height, 768);
        assert_eq!(opts.background_color, "#ffffff");
        assert_eq!(opts.loss_color, "#ff0000");
        assert_eq!(opts.accuracy_color, "#00ff00");
        assert!(!opts.show_sparklines);
    }

    #[test]
    fn test_wasm_dashboard_options_default_trait() {
        let opts = WasmDashboardOptions::default();
        assert_eq!(opts.get_width(), 800);
        assert_eq!(opts.get_height(), 400);
    }
}