学习C2CRS Ⅳ (Conversational Recommender Model)
C2CRS_Model
C2CRS_Model
是一个用于对话推荐系统(Conversational Recommender System, C2CRS)的端到端模型。该模型结合了知识图谱(KG)、上下文信息、用户表示和对话生成等多个模块,以实现高效的推荐和对话功能。它通过以下模块实现:
- 用户表示模块(
CoarseFineDRUserModel
):生成用户表示,结合知识图谱、上下文和评论信息。 - 预训练模块(
CoarseToFinePretrainModel
):通过对比学习进行预训练,提升模型的表示能力。 - 推荐模块(
RecommenderModule
):根据用户表示进行推荐。 - 对话生成模块(
CFSelectionConvModel
):生成对话响应,结合上下文、知识图谱和评论信息。
主要方法与逻辑
1. 初始化方法 (__init__
和 build_model
)
__init__
:- 接收配置对象
opt
、设备
- 接收配置对象