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': {