mirror of
https://github.com/AntoineHX/smart_augmentation.git
synced 2025-05-04 12:10:45 +02:00
modif plot_TF_res
This commit is contained in:
parent
b463ba1b06
commit
ddac630548
3 changed files with 7 additions and 4 deletions
|
@ -45,9 +45,9 @@ if __name__ == "__main__":
|
|||
|
||||
## Acc, Time, Epochs = f(n_tf) ##
|
||||
fig_name="res/TF_nb_tests_compare"
|
||||
inner_its = [10]
|
||||
dataug_epoch_starts= [0]
|
||||
TF_nb = [14]#range(1,14+1)
|
||||
inner_its = [0, 10]
|
||||
dataug_epoch_starts= [0, -1]
|
||||
TF_nb = range(1,14+1)
|
||||
|
||||
fig, ax = plt.subplots(ncols=3, figsize=(30, 8))
|
||||
for in_it in inner_its:
|
||||
|
|
Binary file not shown.
Before Width: | Height: | Size: 27 KiB After Width: | Height: | Size: 214 KiB |
|
@ -114,8 +114,11 @@ def plot_res_compare(filenames, fig_name='res'):
|
|||
|
||||
def plot_TF_res(log, tf_names, fig_name='res'):
|
||||
|
||||
mean = np.mean([x["param"] for x in log], axis=0)
|
||||
std = np.std([x["param"] for x in log], axis=0)
|
||||
|
||||
fig, ax = plt.subplots(1, 1, figsize=(30, 8), sharey=True)
|
||||
ax.bar(tf_names, np.mean([x["param"] for x in log], axis=0), yerr=np.std([x["param"] for x in log], axis=0))
|
||||
ax.bar(tf_names, mean, yerr=std)
|
||||
#ax.bar(tf_names, log[-1]["param"])
|
||||
|
||||
fig_name = fig_name.replace('.',',')
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue