from pprint import pprint import config from builder import buildPlot from utils.file import loadJson from utils.manager import PathManager dataset = 'virushare-20' version = 32 model = 'ProtoNet' report_iter = 100 val_episode = 50 pm = PathManager(dataset, version=version, model_name=model) train_stat_data = loadJson(pm.doc() + 'train_stat.json') train_config = loadJson(pm.doc() + 'train.json') config.reloadArbitraryConfig(train_config, reload_config_list=['plot']) config.plot.Enabled = True vis = buildPlot(config.plot) pprint(train_config) train_metric = np.array(train_stat_data['train']['metrics']).reshape( (-1, report_iter)) # 此处假设metric_num=1 train_loss = np.array(train_stat_data['train']['loss']).reshape( (-1, report_iter)) # validate数据的总数与validate_episode有关 val_metric = np.array(train_stat_data['validate']['metrics']).reshape( (-1, val_episode)) # 此处假设metric_num=1 val_loss = np.array(train_stat_data['validate']['loss']).reshape(
# for ver, conf in tasks.items(): # machine.addTask('train', flatten_update_config=conf) # machine.addTask('test', flatten_update_config={ # 'task|version': ver, # 'load_type': 'best', # }) # machine.addTask('test', flatten_update_config={ # 'task|version': ver, # 'load_type': 'last', # }) s_ver, e_ver = 318, 326 for ver in range(s_ver, e_ver + 1): pm = PathManager(dataset='virushare-20', version=ver) if os.path.exists(pm.doc() + 'test_result.json'): os.remove(pm.doc() + 'test_result.json') machine.addTask('test', flatten_update_config={ 'task|version': ver, 'load_type': 'best', }) machine.addTask('test', flatten_update_config={ 'task|version': ver, 'load_type': 'last', }) # machine.addTask('train', # { # 'task': {