Project Icon

question_extractor

自动从文本数据中提取问答对的开源项目

question_extractor是一个开源项目,能够自动从文本数据中提取问答对。它利用ChatGPT处理大量文档,快速生成可用于训练语言模型的数据集。该项目支持并行处理,可以有效处理长文本,适用于各种文档类型。对于需要构建专业领域问答数据集的研究人员和开发者,question_extractor提供了一个高效、自动化的解决方案。

问题提取器 🧐

大型语言模型可以通过一组问题和答案进行指令调优。 然而,要在自己的数据上进一步微调模型,你需要大量关于你的数据的问题和答案。 手动生成这些问题和答案可能需要大量工作。

这个仓库让你可以使用未经微调的语言模型(ChatGPT)从现有的文本数据中自动提取问题/答案对,消除所有手动工作。

安装

要运行此代码,你需要克隆这个仓库,然后安装以下Python包:

  • tiktoken,OpenAI的分词器,
  • openai,官方OpenAI API客户端,
  • langchain,用于组合模型和实用工具的胶水代码。

使用方法

这个脚本旨在将一个包含markdown(.md)文档的文件夹转换为一个.json文件,其中包含问题、答案和用于生成它们的源文档路径的列表。

要运行代码,请在question_extractor.py文件中设置相关的文件路径(输入文件夹和输出路径),并确保你的OpenAI API密钥在环境中。 然后用Python运行脚本:

python3 question_extractor.py

完成后,所有问题/答案将作为.json文件写入输出路径。

内部工作原理

代码遍历所有文件,对每个文件使用以下提示后跟一段文本来提取问题列表:

你是一位专家用户,正在提取信息以测验人们对文档的了解。你将收到从文档中提取的一页内容,请写出一个编号列表,列出仅基于给定文本可以回答的问题。

然后它遍历这些问题,通过传递以下提示后跟一段文本和一个问题来生成答案:

你是一位回答问题的专家用户。你将收到从文档中提取的一页内容和一个问题。请仅基于给定文本生成一个全面而富有信息的答案。

代码的大部分实际逻辑都用于并发处理文件(以提高速度)并确保传递给模型的文本块足够小,以留出足够的标记用于回答。

如果文本太长而无法发送给模型,它会沿着最高级别的markdown标题进行分割(如果需要,可以重复这个过程,直到拆分到单个段落)。

就性能而言,这个脚本可以在6分钟内处理完整的NERSC文档1。 将318个markdown文件转换为8005个问题,花费29美元。

潜在改进

  • 使用GPT4进行问题回答,以提高答案质量,但会导致运行时间更长和成本显著增加
  • 保存中间结果,以便能够重新启动中断的任务
  • 直接使用OpenAI客户端,而不是Langchain,以减少依赖

Footnotes

  1. 以模型速率限制的约93%运行。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号