augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-23 15:16:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update benchmark_binary.py

Browse Source
This commit is contained in:
augustin64 2023-02-24 11:48:02 +01:00
parent 4df1bf7b5b
commit e1fff4089b
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/scripts/benchmark_binary.py
View File

@ -61,7 +61,7 @@ def compare_binaries(binaries, tries=3, dataset="train"):
results.append(train(binary, base_net, tries, dataset=dataset))
except:
print(f"========== Erreur sur {binary} ==========")
results.append(0)
results.append([0.]*tries)
x = [i for i in range(tries)]
@ -69,9 +69,8 @@ def compare_binaries(binaries, tries=3, dataset="train"):
res = []
for i in range(len(binaries)):
if results[i] != 0:
res.append(ax.plot(x, results[i])[0])
res[i].set_label(binaries[i])
res.append(ax.plot(x, results[i])[0])
res[i].set_label(binaries[i])
ax.set_ylabel("Taux de réussite (%)")
ax.set_xlabel("Nombre de batchs")
@ -80,5 +79,5 @@ def compare_binaries(binaries, tries=3, dataset="train"):
plt.ylim(0, 100)
plt.show()
return results
return binaries, results