transformer-chatbot-pytorch:使用 vanilla 变压器和 GRU 生成多轮对话

时间:2024-06-20 01:40:33
【文件属性】:

文件名称:transformer-chatbot-pytorch:使用 vanilla 变压器和 GRU 生成多轮对话

文件大小:16KB

文件格式:ZIP

更新时间:2024-06-20 01:40:33

nlp natural-language-processing pytorch transformer natural-language-generation

变压器聊天机器人pytorch 这是一个使用ReCoSa 中引入的ReCoSa结构的多轮聊天机器人项目:Detecting the Relevant Contexts with Self-Attention for Multi-turn Dialogue Generation 。 该模型使用 self-attention 机制检测相关对话历史,该机制使用历史级别的转换器编码器,而不是单词级别。 结构的细节如下。 配置 您可以通过修改*目录中的config.json来设置各种参数。 每个变量的描述如下。 (下表中未介绍的均为自动设置,不得更改。) 争论 类型 描述 默认 data_dir String 存储数据文件的父目录的名称。 "data" train_name String 训练数据文件名称的前缀。 "train" valid_name String


【文件预览】:
transformer-chatbot-pytorch-master
----src()
--------main.py(13KB)
--------custom_data.py(5KB)
--------recosa_transformer.py(5KB)
--------data_process.py(12KB)
--------layers.py(7KB)
----requirements.txt(61B)
----.gitignore(2KB)
----README.md(9KB)
----config.json(699B)

网友评论