doc update
[jacinto-ai/caffe-jacinto.git] / plot_top5.py
1 #!/usr/bin/env python
3 import sys
4 import argparse
5 try:
6     import seaborn as sns
7     sns.set(style="dark")
8 except ImportError:
9     pass
10 import matplotlib.pyplot as plt
12 from common_plot import parse_files, plot_accuracy
14 TOP_K = 5
16 parser = argparse.ArgumentParser(description='Plot top {}% from log files.'.format(TOP_K))
17 parser.add_argument('-v', dest='value_at_hover', action='store_true',
18     help="Display plot values at cursor hover")
19 parser.add_argument('-s', dest='separate', action='store_true',
20     help="plot each log separately, don't concatenate them")
21 parser.add_argument('log', nargs = '*', help = "list of log files.")
22 args = parser.parse_args()
24 data = parse_files(files=args.log, top_k=TOP_K, separate=args.separate)
25 plt = plot_accuracy(top_k=TOP_K, data=data, value_at_hover=args.value_at_hover)
27 plt.show()