语言: 中文
标签:

• 代码解释
• 代码摘要
• 摘要生成

许可证: MIT

双子星模型

深入了解我们的模型与方法，请访问博客

模型描述

双子星(Gemini)是基于谷歌T5架构的Transformer模型。该模型在约80万组代码/描述对上进行了预训练，随后在1万条人工合成的高级解释数据上微调。该模型能够为以下编程语言的短至中等长度代码片段生成摘要/解释：

• Python
• JavaScript（主要支持原生JS，但也能处理React等框架）
• Java
• Ruby
• Go

并输出英文描述文本。

应用场景

未经额外微调的双子星模型能以1-2句话解释代码，在Python和JavaScript上表现最佳。我们推荐将其用于：简单代码解释、文档生成，或生产合成数据以优化解释质量。

使用方法

可直接通过文本生成管道调用，如下所示：

from transformers import pipeline, set_seed

summarizer = pipeline('text2text-generation', model='describeai/gemini')
code = "print('hello world!')"

response = summarizer(code, max_length=100, num_beams=3)
print("代码摘要: " + response[0]['generated_text'])

典型输出示例：

代码摘要: 这段代码用于向世界问好。

模型规模

• 双子星(本仓库): 7.7亿参数
• 双子星-精简版: 2.2亿参数

局限性

当前模型可能生成过于简化的描述，无法完整覆盖代码片段功能。我们相信通过更多训练数据可以改善这一现象。

关于我们

Describe.ai致力于构建媲美人类语言理解能力的人工智能系统。虽然前路漫长，我们计划将研究成果通过API贡献给开源社区。