iicd/naswot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add histogram plotting code

Browse Source
This commit is contained in:
jack-willturner
2020-06-17 13:44:25 +01:00
parent 2a1bb3ecc1 76d23b14ac
commit 07a991b3b1
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plot_histograms.py
View File

@@ -60,7 +60,7 @@ def decide_plot(acc, plt_cts, num_rows, boundaries=[60., 70., 80., 90.]):
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Plot histograms of correlation matrix')
parser.add_argument('--data_loc', default='../datasets/cifar/', type=str, help='dataset folder')
parser.add_argument('--api_loc', default='../datasets/NAS-Bench-201-v1_1-096897.pth',
parser.add_argument('--api_loc', default='NAS-Bench-201-v1_1-096897.pth',
type=str, help='path to API')
parser.add_argument('--arch_start', default=0, type=int)
parser.add_argument('--arch_end', default=15625, type=int)