diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 611b13d..59a7ff0 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -119,7 +119,7 @@ def train (**_args): values = _inputhandler._map[key]['values'].tolist() _map[key] = {"beg":beg,"end":end,"values":np.array(values).astype(str).tolist()} info = {"rows":_matrix.shape[0],"cols":_matrix.shape[1],"map":_map} - logger.write({"module":"gan-train","action":"data-prep","context":_args['context'],"input":info}) + logger.write({"module":"gan-train","action":"data-prep","context":_args['context'],"input":_inputhandler._io}) args['logs'] = _args['logs'] if 'logs' in _args else 'logs' args ['max_epochs'] = _args['max_epochs']