Torch_models = ['CFR_MMD',
                'CFR_WASS',
                'BNN',
                'TARNet',
                'DragonNet',
                'ESCFR',
                'DESCN',
                'DeRCFR',
                'Ours1']
SK_models = ['RF',
             'CF',
             'BART',
             'CEVAE',
             'T_learner',
             'S_learner',
             'X_learner',
             'R_learner']

lib_models = ['QHTE',
             'GANITE',
             ]

recommender_models = [
    'NeuMF',
    'NeuMF_IPM',
    'NeuMF_Uncertainty'
]