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

update

Browse Source
This commit is contained in:
HamsterMimi
2024-01-23 10:15:00 +08:00
parent 3f6d16e791
commit 0d830dd2f6
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
correlation/NAS_Bench_201.py
View File

@@ -1,14 +1,11 @@
import argparse import argparse
import os import pickle
import random
import time
from foresight.dataset import * from foresight.dataset import *
from foresight.models import nasbench2 from foresight.models import nasbench2
from foresight.pruners import predictive
from foresight.weight_initializers import init_net from foresight.weight_initializers import init_net
from models import get_cell_based_tiny_net from models import get_cell_based_tiny_net
import pickle
def get_score(net, x, device, measure='meco'): def get_score(net, x, device, measure='meco'):

6
correlation/compute_rho.py
View File

@@ -1,15 +1,15 @@
import pandas as pd import pandas as pd
import pickle import pickle
# path = 'result/sss_cf10_meco.p' path = 'result/sss_cf10_meco_opt.p'
path = 'nb2_sss_cf10_seed42_dlappoint_dlinfo1_initwnone_initbnone_1.p' # path = 'nb2_sss_cf10_seed42_dlappoint_dlinfo1_initwnone_initbnone_1.p'
meco = [] meco = []
accs = [] accs = []
with open(path, 'rb') as f: with open(path, 'rb') as f:
while True: while True:
try: try:
fl = pickle.load(f) fl = pickle.load(f)
meco.append(fl['meco']) meco.append(fl['meco'][0])
accs.append(fl['testacc']) accs.append(fl['testacc'])
except: except:
break break