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

save the scores in swap

Browse Source
This commit is contained in:
Mhrooz 2024-08-25 21:39:15 +02:00
parent 0b66220df4
commit aead4df707
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
correlation.py
View File

@ -93,11 +93,14 @@ if __name__ == "__main__":
print(f'Average SWAP score: {np.mean(swap_score)}')
print(f'Elapsed time: {end_time - start_time:.2f} seconds')
results.append([np.mean(swap_score), acc])
results.append([np.mean(swap_score), acc, i])
results = pd.DataFrame(results, columns=['swap_score', 'valid_acc', 'index'])
results = pd.DataFrame(results, columns=['swap_score', 'valid_acc'])
print()
print(f'Spearman\'s Correlation Coefficient: {stats.spearmanr(results.swap_score, results.valid_acc)[0]}')
results.to_csv('swap_results.csv', float_format='%.4f', index=False)