学习C2CRS Ⅲ (Response Generation Module)
代码地址:https://github.com/RUCAIBox/WSDM2022-C2CRS
论文地址:https://arxiv.org/abs/2201.02732
CFSelectionConvModel模型结构与功能
CFSelectionConvModel
是一个用于对话推荐系统的端到端模型,结合了知识图谱(KG)、评论信息和对话上下文来生成对话响应。它通过以下模块实现:
- 知识图谱模块(
DBModel
):处理知识图谱数据,生成实体表示和上下文表示。 - 评论模块(
CoarseReviewModelForDecoder
):处理评论数据,生成评论表示。 - 对话生成模块(
DecoderCNSelectionModel
):结合知识图谱、评论和对话上下文,生成对话响应。 - 对话上下文编码器:对对话上下文进行编码,生成上下文表示。
主要方法与逻辑
1. 初始化方法 (__init__
和 build_model
)
__init__
:- 接收配