Function create_logger

code/utils.py:128–176

(args, task)

def create_logger(args, task):

    # Logger directory
    os.makedirs(args.log_dir, exist_ok=True)
    os.makedirs(os.path.join(args.log_dir, args.dataset.lower()), exist_ok=True)

    model_info = 'pretrain_{}_{}_{}_{}'.format(args.dataset.lower(), args.kernel, args.num_layer, args.hid_dim)
    if args.single:
        model_info += '_single'
    else:
        model_info += '_{}'.format(args.dec_aggr)

    if args.nonskip:
        model_info += '_nonskip'
    if args.beta != 0:
        model_info += '_beta_{}'.format(str(args.beta))
    if args.norm:
        model_info += '_{}'.format(args.norm)
    
    if task == 'node':
        if args.emb_act:
            model_info += '_emb_act'
        
        log_file = 'log_' + model_info + '.txt'
    elif task == 'graph':
        log_file = 'log_' + model_info + '_pooler_{}.txt'.format(str(args.pooler))

    log_path = os.path.join(args.log_dir, args.dataset.lower(), log_file)
    log_format = '%(levelname)s %(asctime)s - %(message)s'
    log_time_format = '%Y-%m-%d %H:%M:%S'
    
    if args.log:
        log_handlers = [
            logging.FileHandler(log_path),
            logging.StreamHandler(sys.stdout)
        ]
    else:
        log_handlers = [
            logging.StreamHandler(sys.stdout)
        ]
    logging.basicConfig(
        format=log_format,
        datefmt=log_time_format,
        level=logging.INFO,
        handlers=log_handlers
    )
    logger = logging.getLogger()

    return logger, model_info


# ======================================================================