mirror of
https://github.com/AntoineHX/smart_augmentation.git
synced 2025-05-04 20:20:46 +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) ##
|
## Acc, Time, Epochs = f(n_tf) ##
|
||||||
fig_name="res/TF_nb_tests_compare"
|
fig_name="res/TF_nb_tests_compare"
|
||||||
inner_its = [10]
|
inner_its = [0, 10]
|
||||||
dataug_epoch_starts= [0]
|
dataug_epoch_starts= [0, -1]
|
||||||
TF_nb = [14]#range(1,14+1)
|
TF_nb = range(1,14+1)
|
||||||
|
|
||||||
fig, ax = plt.subplots(ncols=3, figsize=(30, 8))
|
fig, ax = plt.subplots(ncols=3, figsize=(30, 8))
|
||||||
for in_it in inner_its:
|
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'):
|
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)
|
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"])
|
#ax.bar(tf_names, log[-1]["param"])
|
||||||
|
|
||||||
fig_name = fig_name.replace('.',',')
|
fig_name = fig_name.replace('.',',')
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue