| |
| import logging |
| import logging.config |
| import PIL |
| import json |
| from os.path import join, exists, split, realpath |
| import time |
| from os import mkdir, getcwd |
| import matplotlib.pyplot as plt |
| import numpy as np |
|
|
|
|
| |
| |
| def get_stamp_from_log(): |
| """ |
| Get the time stamp from the log file |
| :return: |
| """ |
| return split(logging.getLogger().handlers[0].baseFilename)[-1].replace(".log","") |
|
|
|
|
| def create_output_dir(name): |
| """ |
| Create a new directory for outputs, if it does not already exist |
| :param name: (str) the name of the directory |
| :return: the path to the outpur directory |
| """ |
| out_dir = join(getcwd(), name) |
| if not exists(out_dir): |
| mkdir(out_dir) |
| return out_dir |
|
|
|
|
| def init_logger(name_of_model = ""): |
| """ |
| Initialize the logger and create a time stamp for the file |
| """ |
| path = split(realpath(__file__))[0] |
|
|
| with open(join(path, 'log_config.json')) as json_file: |
| log_config_dict = json.load(json_file) |
| |
| |
| filename = ''.join([time.strftime("%y%m%d_%H%M", time.localtime()), ".log"]) |
| |
| |
| log_path = create_output_dir(''.join(['/workspace/HAR_code/out/',name_of_model,'_',filename[:-4], '/'])) |
|
|
| log_config_dict.get('handlers').get('file_handler')['filename'] = join(log_path, ''.join([name_of_model,'_',filename])) |
| logging.config.dictConfig(log_config_dict) |
|
|
| |
| logging.getLogger(PIL.__name__).setLevel(logging.WARNING) |
| |
| return log_path |
|
|
| |
| |
| def plot_loss_func(sample_count, loss_vals, loss_fig_path): |
| plt.figure() |
| plt.plot(sample_count, loss_vals) |
| plt.ylim(0,0.7) |
| plt.grid() |
| plt.title('Loss') |
| plt.xlabel('Epochs') |
| plt.ylabel('Loss') |
| plt.savefig(loss_fig_path) |
|
|
|
|
|
|
