API 参考

本节提供 genrec 框架的详细 API 文档。

核心模块

模型

• RQVAE: 残差量化变分自编码器
• TIGER: 基于 Transformer 的生成式检索模型

数据处理

• 基础数据集: 数据集抽象基类
• 配置管理: 配置管理类
• 处理器: 文本和序列处理工具
• 数据集工厂: 数据集创建工厂模式

训练

• 训练器: 训练工具和脚本
• 模块: 核心构建块（编码器、损失函数、指标）

快速导航

核心组件

模型: - RQVAE 模型类 - 向量量化变分自编码器 - TIGER 模型类 - 基于 Transformer 的生成式检索

数据处理: - 数据集类 - 数据加载和预处理 - 配置系统 - 参数管理 - 处理器 - 文本和序列处理工具

训练: - RQVAE 训练 - 训练流程和配置 - TIGER 训练 - 高级训练工作流

代码示例

查看示例页面了解实用的使用模式和代码片段。