Commit 91d65256 authored by Tommy Duc Luu's avatar Tommy Duc Luu
Browse files


parent 2f4a649c
......@@ -3,14 +3,27 @@ from utils import convert_tb_data
import tensorboard as tb
import matplotlib.pyplot as plt
import os
import time
from colorama import Fore'seaborn-whitegrid')
dirs = ['..\interactive_CA\logs', '..\\final_models',
logdir = dirs[0]
eventdirs = [d for d in os.listdir(logdir)]
dirs = ['data\\growing_results\\samples',
size = ['9x9', '15x15']
type_ = ['carrot', 'rabbit']
colors = ['r','b','g','m','c']
labels = ['ES', 'Adam']
s = size[0]
t = type_[0]
logdir = dirs[0]+os.sep+s+os.sep+t
eventdirs = [d for d in os.listdir(logdir)]
fig = plt.figure()
ax = plt.axes()
ax.set_title(f'{s} {t}')
for dir in eventdirs:
path = f"{logdir}/{dir}"
......@@ -20,8 +33,8 @@ for dir in eventdirs:
print(Fore.RED+f'A directory does not include a tensorboard file:{ve}')
fig = plt.figure()
ax = plt.axes()
ax.plot(df['step'], df['value'])
# plt.savefig(str(np.random.randint)+'.png')
ax.plot(df['step'], df['value'],color=colors.pop(), label=labels.pop())
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment