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_loss
14 parser = argparse.ArgumentParser(description="Plot loss from log files.")
15 parser.add_argument('-v', dest='value_at_hover', action='store_true',
16 help="Display plot values at cursor hover")
17 parser.add_argument('-s', dest='separate', action='store_true',
18 help="plot each log separately, don't concatenate them")
19 parser.add_argument('log', nargs = '*', help = "list of log files.")
20 args = parser.parse_args()
22 data = parse_files(files=args.log, separate=args.separate)
23 plt = plot_loss(data=data, value_at_hover=args.value_at_hover)
25 plt.show()