draw_scalar/
draw_scalar.rs

1use tensorboard_rs::summary_writer::SummaryWriter;
2use std::collections::HashMap;
3
4pub fn main() {
5    let mut writer = SummaryWriter::new(&("./logdir".to_string()));
6
7    let name = "run1";
8    let mut scalar = 2.3;
9    let mut step = 12;    
10    for i in 0..2 {
11        println!("{}", i);
12        scalar += (i as f32)*0.1;
13        step += i;
14
15        writer.add_scalar(name, scalar, step);
16    }
17    writer.flush();
18
19    for n_iter in 0..100 {
20        let mut map = HashMap::new();
21        map.insert("xsinx".to_string(), (n_iter as f32) * (n_iter as f32).sin());
22        map.insert("xcosx".to_string(), (n_iter as f32) * (n_iter as f32).cos());
23        map.insert("arctanx".to_string(), (n_iter as f32).atan());
24        writer.add_scalars("data/scalar_group", &map, n_iter);
25    }
26    writer.flush();
27}