From f72990a675716e702d4e8fb68f3b574bc084e58e Mon Sep 17 00:00:00 2001 From: Mhrooz Date: Mon, 26 Aug 2024 10:53:21 +0200 Subject: [PATCH] update .gitignore file and change the correlation config --- .gitignore | 3 ++- correlation.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index a859fa6..9a838d5 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ __pycache__/ -datasets/ \ No newline at end of file +datasets/ +swap_results.csv \ No newline at end of file diff --git a/correlation.py b/correlation.py index 956d8b6..238ffeb 100644 --- a/correlation.py +++ b/correlation.py @@ -56,7 +56,7 @@ if __name__ == "__main__": results = [] # nasbench_len = 15625 - nasbench_len = 15 + nasbench_len = 15625 # for index, i in arch_info.iterrows(): for i in range(nasbench_len): @@ -99,7 +99,7 @@ if __name__ == "__main__": results.append([np.mean(swap_score), acc, i]) 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(f'Spearman\'s Correlation Coefficient: {stats.spearmanr(results.swap_score, results.valid_acc)[0]}')