--- a +++ b/utils/io.py @@ -0,0 +1,43 @@ +''' +@Author: your name +@Date: 2020-01-06 17:08:52 +@LastEditTime : 2020-01-07 13:25:31 +@LastEditors : Please set LastEditors +@Description: In User Settings Edit +@FilePath: /KGCN_Keras-master/utils/io.py +''' +# -*- coding: utf-8 -*- + +import os +import json +import pickle + + +def pickle_load(filename: str): + try: + with open(filename, 'rb') as f: + obj = pickle.load(f) + print(f'Logging Info - Loaded: {filename}') + except EOFError: + print(f'Logging Error - Cannot load: {filename}') + obj = None + + return obj + + +def pickle_dump(filename: str, obj): + with open(filename, 'wb') as f: + pickle.dump(obj, f) + print(f'Logging Info - Saved: {filename}') + + +def write_log(filename: str, log, mode='w'): + with open(filename, mode) as writers: + writers.write('\n') + json.dump(log, writers, indent=4, ensure_ascii=False) + + +def format_filename(_dir: str, filename_template: str, **kwargs): + """Obtain the filename of data base on the provided template and parameters""" + filename = os.path.join(_dir, filename_template.format(**kwargs)) + return filename