From aead4df707ee0bfb5d5c95c0d945e1d97ef0e8e1 Mon Sep 17 00:00:00 2001 From: Mhrooz Date: Sun, 25 Aug 2024 21:39:15 +0200 Subject: [PATCH] save the scores in swap --- correlation.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/correlation.py b/correlation.py index 4538354..fbaa828 100644 --- a/correlation.py +++ b/correlation.py @@ -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) +