|
|
|
@ -144,9 +144,12 @@ def generate(**_args):
|
|
|
|
|
"""
|
|
|
|
|
partition = _args['partition'] if 'partition' in _args else None
|
|
|
|
|
if not partition :
|
|
|
|
|
f = open(os.sep.join([_args['logs'],'output',_args['context'],'map.json']))
|
|
|
|
|
LOG_DIR = os.sep.join([_args['logs'],'output',_args['context']])
|
|
|
|
|
# f = open(os.sep.join([_args['logs'],'output',_args['context'],'map.json']))
|
|
|
|
|
else:
|
|
|
|
|
f = open(os.sep.join([_args['logs'],'output',_args['context'],str(partition),'map.json']))
|
|
|
|
|
LOG_DIR = os.sep.join([_args['logs'],'output',_args['context'],str(partition)])
|
|
|
|
|
# f = open(os.sep.join([_args['logs'],'output',_args['context'],str(partition),'map.json']))
|
|
|
|
|
f = open(os.sep.join([LOG_DIR,'map.json']))
|
|
|
|
|
_map = json.loads(f.read())
|
|
|
|
|
f.close()
|
|
|
|
|
# if 'file' in _args :
|
|
|
|
@ -154,7 +157,7 @@ def generate(**_args):
|
|
|
|
|
# else:
|
|
|
|
|
# df = _args['data'] if not isinstance(_args['data'],str) else pd.read_csv(_args['data'])
|
|
|
|
|
args = {"context":_args['context'],"max_epochs":_args['max_epochs'],"candidates":_args['candidates']}
|
|
|
|
|
args['logs'] = _args['logs'] if 'logs' in _args else 'logs'
|
|
|
|
|
args['logs'] = LOG_DIR if 'logs' in _args else 'logs'
|
|
|
|
|
args ['max_epochs'] = _args['max_epochs']
|
|
|
|
|
# args['matrix_size'] = _matrix.shape[0]
|
|
|
|
|
args['batch_size'] = 2000
|
|
|
|
|