From c80cfb8cac4b5ca398db5697728f97986dcba937 Mon Sep 17 00:00:00 2001 From: Mhrooz Date: Thu, 29 Aug 2024 09:20:29 +0200 Subject: [PATCH] add parser --- analyze.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/analyze.py b/analyze.py index d809373..9e7f121 100644 --- a/analyze.py +++ b/analyze.py @@ -2,13 +2,15 @@ import csv import matplotlib.pyplot as plt from scipy import stats import pandas as pd +import argparse -def plot(l): +def plot(l,filename): + threshold = [0, 10000, 20000, 30000, 40000, 50000, 60000, 70000] labels = ['0-10k', '10k-20k,', '20k-30k', '30k-40k', '40k-50k', '50k-60k', '60k-70k'] l = [i/15625 for i in l] l = l[:7] plt.bar(labels, l) - plt.savefig('plot.png') + plt.savefig(filename) def analyse(filename): l = [0 for i in range(10)] @@ -35,11 +37,16 @@ def analyse(filename): results = pd.DataFrame(datas, columns=['swap_score', 'valid_acc', 'index']) print(results['swap_score'].max()) print(best_value) - plot(l) + plot(l, filename + '.png') return stats.spearmanr(results.swap_score, results.valid_acc)[0] if __name__ == '__main__': - print(analyse('output/swap_results.csv')) + parser = argparse.ArgumentParser() + parser.add_argument('--filename', type=str, help='Filename to analyze', default='swap_results.csv') + + args = parser.parse_args() + + print(analyse('output' + '/' + args.filename))