3 releases
0.5.7 | Mar 12, 2022 |
---|---|
0.5.6 | Mar 7, 2022 |
0.5.5 | Mar 7, 2022 |
#391 in Machine learning
108 downloads per month
Used in 5 crates
(via tensorboard-rs)
765KB
16K
SLoC
Write to tensorboard in Rust
Write TensorBoard events in Rust.
- Can write
scalar
,image
,histogram
.
Example
- Write multiple scalar in one plot.
let mut writer = SummaryWriter::new(&("./logdir".to_string()));
for n_iter in 0..100 {
let mut map = HashMap::new();
map.insert("x1".to_string(), (n_iter as f32));
map.insert("x^2".to_string(), (n_iter as f32) * (n_iter as f32));
writer.add_scalars("data/scalar_group", &map, n_iter);
}
writer.flush();
Dependencies
~1.5MB
~28K SLoC