save the scores in swap
This commit is contained in:
parent
0b66220df4
commit
aead4df707
@ -93,11 +93,14 @@ if __name__ == "__main__":
|
|||||||
print(f'Average SWAP score: {np.mean(swap_score)}')
|
print(f'Average SWAP score: {np.mean(swap_score)}')
|
||||||
print(f'Elapsed time: {end_time - start_time:.2f} seconds')
|
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()
|
||||||
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]}')
|
||||||
|
results.to_csv('swap_results.csv', float_format='%.4f', index=False)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user