# -*- coding: utf-8 -*-
# @Time : 2020/8/7
# @Author : Zihan Lin
# @Email : linzihan.super@foxmail.com
"""
recbole.utils.logger
###############################
"""
import logging
import os
from recbole.utils.utils import get_local_time
[docs]def init_logger(config):
"""
A logger that can show a message on standard output and write it into the
file named `filename` simultaneously.
All the message that you want to log MUST be str.
Args:
config (Config): An instance object of Config, used to record parameter information.
Example:
>>> logger = logging.getLogger(config)
>>> logger.debug(train_state)
>>> logger.info(train_result)
"""
LOGROOT = './log/'
dir_name = os.path.dirname(LOGROOT)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
logfilename = '{}-{}.log'.format(config['model'], get_local_time())
logfilepath = os.path.join(LOGROOT, logfilename)
filefmt = "%(asctime)-15s %(levelname)s %(message)s"
filedatefmt = "%a %d %b %Y %H:%M:%S"
fileformatter = logging.Formatter(filefmt, filedatefmt)
sfmt = "%(asctime)-15s %(levelname)s %(message)s"
sdatefmt = "%d %b %H:%M"
sformatter = logging.Formatter(sfmt, sdatefmt)
if config['state'] is None or config['state'].lower() == 'info':
level = logging.INFO
elif config['state'].lower() == 'debug':
level = logging.DEBUG
elif config['state'].lower() == 'error':
level = logging.ERROR
elif config['state'].lower() == 'warning':
level = logging.WARNING
elif config['state'].lower() == 'critical':
level = logging.CRITICAL
else:
level = logging.INFO
fh = logging.FileHandler(logfilepath)
fh.setLevel(level)
fh.setFormatter(fileformatter)
sh = logging.StreamHandler()
sh.setLevel(level)
sh.setFormatter(sformatter)
logging.basicConfig(
level=level,
handlers=[fh, sh]
)