run_gen_item_emb.py

# @Time   : 2020/7/20
# @Author : Shanlei Mu
# @Email  : slmu@ruc.edu.cn

# UPDATE
# @Time   : 2020/10/3, 2020/10/1
# @Author : Yupeng Hou, Zihan Lin
# @Email  : houyupeng@ruc.edu.cn, zhlin@ruc.edu.cn


import argparse

from recbole.quick_start import run_emb_gen_train


if __name__ == '__main__':
    try:
        parser = argparse.ArgumentParser()
        parser.add_argument('--model', '-m', type=str, default='BPR', help='name of models')
        parser.add_argument('--dataset', '-d', type=str, default='ml-100k', help='name of datasets')
        parser.add_argument('--config_files', type=str, default=None, help='config files')
        parser.add_argument('--saved', type=str, default='True', help='saved')
        parser.add_argument('--hint', type=str, default='', help='hint for run_recbole')

        args, _ = parser.parse_known_args()

        config_file_list = args.config_files.strip().split(',') if args.config_files else None
        saved = (args.saved.lower() == 'true')
        result = run_emb_gen_train(model=args.model, dataset=args.dataset, config_file_list=config_file_list, saved=saved)
        try:
            from mtjupyter_utils import remind
            message = ' '.join([str(args.model), args.hint]) + '\n' + f'model_file = {result}'
            remind(message)
        except Exception as e:
            pass
    except Exception as err:
        try:
            from mtjupyter_utils import remind
            remind(str(err))
            raise err
        except Exception as e:
            raise err