tf2-transformer-chatbot

用TensorFlow 2快速上手Transformer聊天机器人,支持TPU训练。

这是一个基于TensorFlow 2实现的Transformer聊天机器人项目,支持TPU训练。它利用Transformer架构(自注意力机制)构建对话模型,解决了传统序列到序列模型在长距离依赖和并行计算上的不足。项目提供了完整的训练和推理流程,包括数据预处理、模型定义、训练循环和交互式对话脚本。核心能力包括:支持英法翻译任务作为示例,展示Transformer在生成任务上的应用;提供TPU/GPU训练支持,适合大规模实验;代码结构清晰,适合学习Transformer在对话生成中的实践。项目以Jupyter Notebook为主要形式,便于逐步理解和修改,但工程化程度较低,更适合研究或教学场景。

开源 free 对话助手
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 130
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类对话助手
开发团队bryanlimy
所属国家
官网地址
定价模式free
价格说明开源项目,MIT许可证,完全免费,需自行部署。
访问状态
是否开源是
开源协议MIT
主要语言Jupyter Notebook
技术栈/模型chatbot,deep-learning,tensorflow,tensorflow-2,tensorflow-tutorials,tensorflow2,transformer
GitHub 星标★ 130
30天Star增速
HF 下载量
上线时间2019-04-14 00:00:00
最近更新2026-06-24 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数3

使用教程

核心亮点

  • 代码简洁,基于TF 2原生API,易于理解和二次开发
  • 提供TPU训练支持,适合大规模实验
  • 以翻译任务示例,清晰展示Transformer生成流程

不足之处

  • 示例任务为翻译而非真正开放域对话,对话能力有限
  • 依赖较老的数据集和预处理,可能需自行适配新数据
  • 文档/社区待观察

适用场景

  • 学习Transformer架构在生成任务中的实现
  • 快速搭建一个基于翻译的聊天原型
  • 在TPU上实验Transformer训练效果

替代项目

tensorflow/models official transformer、Hugging Face Transformers、DialoGPT

项目介绍

tf2-transformer-chatbot 是对话助手领域的开源项目,由 bryanlimy 开发,2019 年首次发布。

它收录于对话助手分类,该分类目前共有 907 个项目,GitHub 星标数为 130。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-06-24。MIT许可证,完全免费,需自行部署。

它主要面向的使用场景是:学习Transformer架构在生成任务中的实现。同类可对比的替代方案包括 tensorflow/models official transformer、Hugging Face Transformers、DialoGPT。

上一篇:DocNavigator

下一篇:xiaozhi-webrtc

同类项目推荐

open-webui 开源

给本地模型配个漂亮聊天室,全家都能用

User-friendly AI Interface (Supports Ollama, OpenAI API, ...)

★ 152821 2026-08-09
prompts.chat 开源

海量好提示词,抄了就能让 AI 更听话

f.k.a. Awesome ChatGPT Prompts. Share, discover, and collect prompts from the commun···

★ 170999 2026-08-09
elia 开源

终端里用键盘快速聊 AI,多模型切换不打断思路

A snappy, keyboard-centric terminal user interface for interacting with large langua···

★ 2479 2026-08-09
NextChat 开源

一个界面聊遍所有主流AI模型,支持全平台部署,轻快又私密。

✨ Zero-config AI chat assistant. No API key needed — sign up and instantly chat wi···

★ 88802 2026-08-09