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

update .gitignore file and change the correlation config

Browse Source
This commit is contained in:
Mhrooz
2024-08-26 10:53:21 +02:00
parent ff85bba9cd
commit f72990a675
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -1,2 +1,3 @@
__pycache__/ __pycache__/
datasets/ datasets/
swap_results.csv

4
correlation.py
View File

@@ -56,7 +56,7 @@ if __name__ == "__main__":
results = [] results = []
# nasbench_len = 15625 # nasbench_len = 15625
nasbench_len = 15 nasbench_len = 15625
# for index, i in arch_info.iterrows(): # for index, i in arch_info.iterrows():
for i in range(nasbench_len): for i in range(nasbench_len):
@@ -99,7 +99,7 @@ if __name__ == "__main__":
results.append([np.mean(swap_score), acc, i]) 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', 'index'])
results.to_csv('swap_results.csv', float_format='%.4f', index=False) results.to_csv('output/swap_results.csv', float_format='%.4f', index=False)
print() print()
print(f'Spearman\'s Correlation Coefficient: {stats.spearmanr(results.swap_score, results.valid_acc)[0]}') print(f'Spearman\'s Correlation Coefficient: {stats.spearmanr(results.swap_score, results.valid_acc)[0]}')