a0c/visualize.py at master · tmoer/a0c · GitHub

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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Wrapper for visualize.py

@author: thomas
"""

if __name__ == '__main__' and __package__ is None:
    from os import sys, path
    sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))

import os
import argparse
from common.visualize import loop_directories
from hps import get_hps


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--folder', default='/home/thomas/a0c_results')
    parser.add_argument('--home', action='store_true',default=False)
    parser.add_argument('--game', default=None)
    parser.add_argument('--name', default=None)
    parser.add_argument('--subindex', default=None)
    parser.add_argument('--plot_type', default='lc')
    parser.add_argument('--window', type=int,default=25)
    parser.add_argument('--sd', action='store_true',default=False)
    parser.add_argument('--on_mean', action='store_true',default=False)
    parser.add_argument('--item', default='return',help='This item in result will be plotted')
    parser.add_argument('--remove', action='store_true',default=False)
    parser.add_argument('--plotlen', type=int,default=25)
    parser.add_argument('--xlim', nargs='+',type=float,default=None)
    parser.add_argument('--ylim', nargs='+',type=float,default=None)
    parser.add_argument('--errlim', nargs='+',type=float,default=None,help='Limits on the errorbars')
    parser.add_argument('--item1_label', nargs='+', default=None)
    parser.add_argument('--item2_label', nargs='+', default=None)
    parser.add_argument('--item3_label', nargs='+', default=None)
    parser.add_argument('--no_suptitle', action='store_true',default=False)
    parser.add_argument('--x_item', default='steps') # steps or eps

    parser.add_argument('--line_item', default='item1') #
    parser.add_argument('--col_item', default='item2') #
    parser.add_argument('--row_item', default='item3') #

    args = parser.parse_args()

    if args.item1_label is not None: args.item1_label = ' '.join(args.item1_label)
    if args.item2_label is not None: args.item2_label = ' '.join(args.item2_label)
    if args.item3_label is not None: args.item3_label = ' '.join(args.item3_label)

    if args.home:
        result_folder = os.getcwd() + '/results/'
    else:
        result_folder = args.folder + '/'
    print('Start processing folder {}'.format(result_folder))
    overview_dir= result_folder+'learning_curves/'
    if not os.path.exists(overview_dir):
        os.makedirs(overview_dir)

    loop_directories(result_folder=result_folder,overview_dir=overview_dir,game=args.game,name=args.name,
                     subindex=args.subindex,plot_type=args.plot_type,window=args.window,sd=args.sd,on_mean=args.on_mean,
                     item=args.item,remove=args.remove,plotlen=args.plotlen,xlim=args.xlim,ylim=args.ylim,errlim=args.errlim,
                     get_hps=get_hps,no_suptitle=args.no_suptitle,x_item=args.x_item,line_item=args.line_item,col_item=args.col_item,
                     row_item=args.row_item,item1_label=args.item1_label,item2_label=args.item2_label,item3_label=args.item3_label)