File size: 1,529 Bytes
c508d7f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from numpy.random import rand
from tensorboardX import SummaryWriter
import time


with SummaryWriter() as writer:
    for n_iter in range(100):
        writer.add_scalar('twse/0050', rand(), n_iter)
        writer.add_scalar('twse/2330', rand(), n_iter)
        t = rand()
        writer.add_scalar('dow/aaa', t, n_iter)
        writer.add_scalar('dow/bbb', t - 1, n_iter)
        writer.add_scalar('dow/ccc', t + 1, n_iter)
        writer.add_scalar('nasdaq/aaa', rand(), n_iter)
        writer.add_scalar('nasdaq/bbb', rand(), n_iter)
        writer.add_scalar('nasdaq/ccc', rand(), n_iter)

    layout = {'Taiwan': {'twse': ['Multiline', ['twse/0050', 'twse/2330']]},
              'USA': {'dow': ['Margin', ['dow/aaa', 'dow/bbb', 'dow/ccc']],
                      'nasdaq': ['Margin', ['nasdaq/aaa', 'nasdaq/bbb', 'nasdaq/ccc']]}}
    writer.add_custom_scalars(layout)
#    writer.add_custom_scalars(layout) second call has no effect

time.sleep(1)

with SummaryWriter() as writer:
    for n_iter in range(100):
        writer.add_scalar('twse/0050', rand(), n_iter)
        writer.add_scalar('twse/2330', rand(), n_iter)

    writer.add_custom_scalars_multilinechart(['twse/0050', 'twse/2330'])

time.sleep(1)

with SummaryWriter() as writer:
    for n_iter in range(100):
        t = rand()
        writer.add_scalar('dow/aaa', t, n_iter)
        writer.add_scalar('dow/bbb', t - 1, n_iter)
        writer.add_scalar('dow/ccc', t + 1, n_iter)

    writer.add_custom_scalars_marginchart(['dow/aaa', 'dow/bbb', 'dow/ccc'])