iicd/swap-nas
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add parser

Browse Source
This commit is contained in:
Mhrooz
2024-08-29 09:20:29 +02:00
parent 551abc31f3
commit c80cfb8cac
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
analyze.py
View File

@@ -2,13 +2,15 @@ import csv
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy import stats from scipy import stats
import pandas as pd import pandas as pd
import argparse
def plot(l): def plot(l,filename):
threshold = [0, 10000, 20000, 30000, 40000, 50000, 60000, 70000]
labels = ['0-10k', '10k-20k,', '20k-30k', '30k-40k', '40k-50k', '50k-60k', '60k-70k'] labels = ['0-10k', '10k-20k,', '20k-30k', '30k-40k', '40k-50k', '50k-60k', '60k-70k']
l = [i/15625 for i in l] l = [i/15625 for i in l]
l = l[:7] l = l[:7]
plt.bar(labels, l) plt.bar(labels, l)
plt.savefig('plot.png') plt.savefig(filename)
def analyse(filename): def analyse(filename):
l = [0 for i in range(10)] l = [0 for i in range(10)]
@@ -35,11 +37,16 @@ def analyse(filename):
results = pd.DataFrame(datas, columns=['swap_score', 'valid_acc', 'index']) results = pd.DataFrame(datas, columns=['swap_score', 'valid_acc', 'index'])
print(results['swap_score'].max()) print(results['swap_score'].max())
print(best_value) print(best_value)
plot(l) plot(l, filename + '.png')
return stats.spearmanr(results.swap_score, results.valid_acc)[0] return stats.spearmanr(results.swap_score, results.valid_acc)[0]
if __name__ == '__main__': if __name__ == '__main__':
print(analyse('output/swap_results.csv')) parser = argparse.ArgumentParser()
parser.add_argument('--filename', type=str, help='Filename to analyze', default='swap_results.csv')
args = parser.parse_args()
print(analyse('output' + '/' + args.filename))