在构建text-to-sql
模型时,高质量的数据和有效的数据流程是必不可少的。目前市面上已经有许多优秀的开源大模型,如ChatLLaMa
、Alpaca
、Vicuna
、以及Databricks-Dolly
,Stable Diffution
母公司发布的StableLM
等
此外,还有一些训练框架可供选择,比如LMFlow
和微软最近开源的DeepSpeed
等
但即使开源的大模型和训练框架都越来越多,它们也都离不开高质量数据和生产高质量数据的流程,这也是一切模型构建的前提;这篇文章就来讲一下在私有项目中我是如何持续积累高质量数据集,并沉淀相关流程框架的
prompt
的方法论 / 框架,通常是规定 / 限制从模型中获得所需输出的框架,并持续的通过实验和分析来优化这些prompt
Prompt Engineering
很重要,因为它可以显著影响NLP模型的性能,特别是对于微调任务。设计良好的prompt
可以使模型生成更准确和相关的响应,而设计不当的提示可能会导致不准确或无用的输出,Prompt Engineering
能够帮助我们进行LLM对接、构建和理解其功能,并提高其安全性
Prompt Engineering
的常见技术包括限制prompt
的长度和结构、合并上下文和背景知识,以及使用各种类型的提示,如填空式prompt
、多项选择或自由格式文本prompt
在开源社区中存在着许多文本到text-to-sql
数据集,包括但不限于WikiSQL, SParC, Spider, HybridSQL, CoSQL等。这些数据集可以作为模型的训练集、验证集和测试集,并且其高质量的标注使得它们有资格评价模型性能的标准,这些开源数据集通常也维护了一份LeaderBoard
来show
出使用它们训练出的模型表现,如下是我收集的来自huggingface
, paperswithcode
, Github
的一些优质的text-to-sql
数据集:
这些数据集经过一些格式处理和信息补充,即可直接作为我们模型的训练数据集;格式处理过程这里不做过多阐述,完成格式处理后,还需经过数据校验以及SQL分析,这些功能由下面介绍的Prompt-Collector提供
除了上面提到过的方式——使用已有的开源数据集,我们还可以通过基于GPT-3.5
这个已经“大成”的LLM chatbot生成自定义数据集。而为了大量的生产包括但不限于text-to-sql
场景的高质量数据集,我们首先需要一套完整的数据集生产框架,囊括数据生成,生成后的数据分析,数据验证,以及生成结构化的Prompt等功能
这里我基于cobra提供的创建类git / go tools命令行工具的能力,实现了提供一整套数据集生成能力的命令行工具Prompt-Collector
,以下是Prompt-Collector
的架构图:
首先我们来看看最基本但也是最重要的数据集生成,这里的数据集生成是指通过prompt
让GPT-3.5模拟LLM instruction completion接口
的输入和输出来生成数据集;在这个子步骤中我们需要使用到一些prompt engineering
的技巧来设计生成数据的方式,这里我简单介绍一些prompt engineering
的基础内容,详细内容我会在后续文章中展开阐述:
LLM
更倾向于可能更高的下一个token
;而Temperature升高可能会导致更多的随机性(换句话说也可以是更多的创造性),这实际上是在增加其他可能token
的权重
在应用方面,我们可以对基于事实的类QA任务
使用较低的Temperature,以确保其返回确切的事实和简洁的回答;而对于诗歌创作或其他Idea型任务
,我们则可以选择适当地增大Temperature(其值域为0到2)以获取更多创造性的输出
我们当前获取数据集时设置为0.3
,能够比较稳定的输出标准格式的数据并且能够输出比较多样化的Schema和Query
⚠️注意:一般来说只调整Temperature就够了,不需要和类似的另一个参数Top_p一起调整prompt
中只包含直接的问题或是说明(没有任何希望它完成的任务的示例或演示)
<Question>? / <Instruction>
相对应的,few-shot prompting指的是提供少量希望LLM输出的示例,如经典的QA格式:
Q: <Question>?A: <Answer>Q: <Question>?A: <Answer>Q: <Question>?A: <Answer>Q: <Question>?A:prompt
构成:
JSON Format
等)基于以上构成,我们就可以具体描述大部分任务,并指示LLM
更有效地返回内容,当然,也不是所有问题都需要这完整的四部分,仅仅一条简短的Instruction也可以收获到高质量的response
基于上面提到过的prompt
构建技巧和框架,最终我们就可以得出一个比较有效的prompt
了,如以下是我在获取针对clickhouse
数据源的text-to-sql
数据集时初步使用的一个prompt
:
## InstructionGenerate a dataset## Input DataGive me %d specific cases of Clickhouse high-performance query scenarios, it should be returned in a List.Every case item contains 3 fields, field "schema" is the table structure(CREATE TABLE SQL), field "question" is the natural language question, and field "query" is the SQL query for the question.The query need to include a combination of the following Clickhouse capabilities: a. Clickhouse functions like rank、dense_rank,、row_number、stochasticLogisticRegression、evalMLMethod、stochasticLinearRegression、arrayPopBack、arrayPopFront、arrayPushBack、arrayPushFront、arrayResize、arraySlice、arrayJoin、arrayDifference、arrayDistinct、tumble、hop、tumbleStart等Time Window、L1Norm、L2Norm、LinfNorm b. Use Aggregate functions such as group by, ROLL UP, CUBE, etc. c. Use Where and Pre Where to filter time, product, etc. d. Use select syntax such as With, Sample, OFFSET, LIMIT, etc. e. You can build complex scenarios with multiple with clauses / sub queries f. Please make sure it is the best in terms of performance## Output IndicatorPure JSON Format string with no quotes, no explanations, no self-reference, no apologies, just answer.For example, this is a dataset contains 2 cases:` ``` `[ { "schema": "CREATE TABLE IF NOT EXISTS product_recommendation_model( customer_id UInt32, recommendation Array(UInt32), PRIMARY KEY (customer_id)) ENGINE = MergeTree() ORDER BY customer_id;", "question": "Recommend 5 products with similar purchase history to customer 123.", "query": "WITH( SELECT customer_id, recommendation, row_number() OVER ( ORDER BY L2Norm(recommendation) DESC) AS row_num FROM ( SELECT customer_id, arrayMap( (i, score) -> IF(score > 0.5, i, 0), arrayEnumerate(predictor) ) AS recommendationFROM customer_product_stateWHERE date >= '2022-01-01' AND date <= '2022-02-01' AND customer_id = 123 ) GROUP BY customer_id, recommendation ) AS recommendation_ranksINSERT INTO product_recommendation_model (customer_id, recommendation) SELECT customer_id, recommendationFROM recommendation_ranksWHERE row_num <= 5;" }, { "schema": "CREATE OR REPLACE TABLE hits(WatchID BIGINT, JavaEnable SMALLINT, Title TEXT, GoodEvent SMALLINT, EventTime TIMESTAMP, EventDate Date, CounterID INTEGER, ClientIP INTEGER, RegionID INTEGER, UserID BIGINT, CounterClass SMALLINT, OS SMALLINT, UserAgent SMALLINT, URL TEXT, Referer TEXT, IsRefresh SMALLINT, RefererCategoryID SMALLINT, RefererRegionID INTEGER, URLCategoryID SMALLINT, URLRegionID INTEGER, ResolutionWidth SMALLINT, ResolutionHeight SMALLINT, ResolutionDepth SMALLINT, FlashMajor SMALLINT, FlashMinor SMALLINT, FlashMinor2 TEXT, NetMajor SMALLINT, NetMinor SMALLINT, UserAgentMajor SMALLINT, UserAgentMinor VARCHAR(255), CookieEnable SMALLINT, JavascriptEnable SMALLINT, IsMobile SMALLINT, MobilePhone SMALLINT, MobilePhoneModel TEXT, Params TEXT, IPNetworkID INTEGER, TraficSourceID SMALLINT, SearchEngineID SMALLINT, SearchPhrase TEXT, AdvEngineID SMALLINT, IsArtifical SMALLINT, WindowClientWidth SMALLINT, WindowClientHeight SMALLINT, ClientTimeZone SMALLINT, ClientEventTime TIMESTAMP, SilverlightVersion1 SMALLINT, SilverlightVersion2 SMALLINT, SilverlightVersion3 INTEGER, SilverlightVersion4 SMALLINT, PageCharset TEXT, CodeVersion INTEGER, IsLink SMALLINT, IsDownload SMALLINT, IsNotBounce SMALLINT, FUniqID BIGINT, OriginalURL TEXT, HID INTEGER, IsOldCounter SMALLINT, IsEvent SMALLINT, IsParameter SMALLINT, DontCountHits SMALLINT, WithHash SMALLINT, HitColor CHAR, LocalEventTime TIMESTAMP, Age SMALLINT, Sex SMALLINT, Income SMALLINT, Interests SMALLINT, Robotness SMALLINT, RemoteIP INTEGER, WindowName INTEGER, OpenerName INTEGER, HistoryLength SMALLINT, BrowserLanguage TEXT, BrowserCountry TEXT, SocialNetwork TEXT, SocialAction TEXT, HTTPError SMALLINT, SendTiming INTEGER, DNSTiming INTEGER, ConnectTiming INTEGER, ResponseStartTiming INTEGER, ResponseEndTiming INTEGER, FetchTiming INTEGER, SocialSourceNetworkID SMALLINT, SocialSourcePage TEXT, ParamPrice BIGINT, ParamOrderID TEXT, ParamCurrency TEXT, ParamCurrencyID SMALLINT, OpenstatServiceName TEXT, OpenstatCampaignID TEXT, OpenstatAdID TEXT, OpenstatSourceID TEXT, UTMSource TEXT, UTMMedium TEXT, UTMCampaign TEXT, UTMContent TEXT, UTMTerm TEXT, FromTag TEXT, HasGCLID SMALLINT, RefererHash BIGINT, URLHash BIGINT, CLID INTEGER) ENGINE = MergeTree() ORDER BY (CounterID, EventDate, UserID, EventTime, WatchID)", "question": "give me page view count bucketed by date and order by date between 2013-07-14 and 2013-07-15 where CounterID is 62 and is not refreshed", "query": "SELECT DATE_TRUNC('day', EventTime) AS M, COUNT(*) AS PageViews FROM hits WHERE CounterID = 62 AND EventDate >= '2013-07-14' AND EventDate <= '2013-07-15' AND IsRefresh = 0 GROUP BY DATE_TRUNC('day', EventTime) ORDER BY DATE_TRUNC('day', EventTime)" }]` ``` `
这个版本的prompt
包含了Instruction, Input Data 以及 Output Indicator,在能够保证输出格式稳定为pure JSON
的同时,也能比较好的覆盖到全部的clickhouse
函数使用场景,且在后续的Validate
中达到较高的留存率
当然,prompt
构建是一个迭代过程,需要进行大量试验才能获得有效且稳定的结果,这并不是一蹴而就的;这份prompt
在后续也被我拆解成了成更简单的子任务prompt
,并针对其返回结果进行持续优化。在prompt
构建的初期并不建议写大段的prompt,而是应该从简单的prompt
开始,不断添加更多的元素和上下文,以获得更好的结果,其实很多简单的prompt
也能收获很好的高质量response
而当我们有一个涉及许多不同子任务的大型任务时,我们可以试着把任务分解成更简单的子任务(分步骤 / 分问题),分别完成逐步的prompt
构建,不断积累;在子任务都能获取到稳定的response
后,再尝试合并;这就避免了在项目前期给prompt
设计过程增加太多的复杂性
其余的一些prompt
设计的技巧和注意事项我会在另一篇Prompt-Engineering
的专题文章中介绍,这里不继续展开
准备好生产text-to-sql dataset
的prompt
后,我们就需要真正地调用OpenAI API
生产数据集了,这里我们使用go-openai管理OpenAI client以及调用API
首先初始化Client:
client := openai.NewClient(token)
client初始化完成后,设置好model, messages, maxToken以及temperature(上面介绍过的关键参数,这里设置为0.3),调用其CreateChatCompletion
方法即可:
// Create a new instance of the CompletionRequest struct// Set up the chat parameters(<https://platform.openai.com/docs/api-reference/completions>)params := openai.ChatCompletionRequest{ Model: openai.GPT3Dot5Turbo, Messages: append(messages, openai.ChatCompletionMessage{ Role: openai.ChatMessageRoleUser, Content: newPrompt, }), MaxTokens: DefaultMaxTokens, Temperature: DefaultTemperature,}response, err = client.CreateChatCompletion(ctx, params)if err != nil { err = fmt.Errorf("[Chat] failed to create chat completion: %w", err)}
成功拿到response
后,即可直接从response.Choices[0].Message.Content
获取到返回的内容,再进行进一步解析(同时我们还可以从resp.Usage.TotalTokens
获取到该次请求消耗的token
数)
在进行实际请求时,我们还需要处理以下两种异常情况:
OpenAI
的API都有请求频率限制(官方文档),尤其是Free Granted API Key
的Rate Limit
,目前为3次每分钟,见下图,RPM指每分钟请求限制,TPM指每分钟token限制),而我们的token池包含了部分免费token(压低总体成本以及填充其它key的Rate Limit);基于这种现状我就需要构建一个限制对应频率的OpenAI client池来访问OpenAI APIOpenAI
代理服务token,虽然代理服务解决了大陆地区访问OpenAI API
导致封号的问题,但由于代理服务器的稳定性等问题,也会出现偶现的504,这时可以进行简单的重试,也可以在重试一定次数无果后更换为非代理client进行访问完成了异常请求的处理后,我们就可以稳定的获取数据了
这里我们前期选择使用jsonl
格式直接存储数据(一行是一份数据),方便生成数据时并发写入文件,以及在超时控制 / 异常处理时可以直接中断任务上传结果数据集文件到构件 / 仓库等
至此我们成功从GPT-3.5-turbo
拿到了想要的text-to-sql dataset
,但为了保证生成数据的质量,我们还需要对数据进行校验
获得GPT-3.5
生成的数据之后,无疑我们仍然需要进行严格的数据校验
在text-to-sql
模型中,常见的数据错误包括但不限于问题理解歧义、缺失信息、语言表述不准确、甚至更严重的SQL无法执行等问题。为了避免这些问题,在构建文本到SQL数据集时,应该尽可能地消除这些数据错误,以确保训练出来的模型表现最佳
而在这些问题中,我们可以通过自动化校验消除SQL无法执行的错误
一份 AI-SQL 训练数据结构如下:
type ( AiSQLData struct { Datasource string `json:"datasource"` // input part of text-to-sql model Question string `json:"question"` Schema string `json:"schema"` // output part of text-to-sql model AiSQLOutput } AiSQLOutput struct { Query string `json:"query"` Analysis string `json:"analysis"` Table [][]any `json:"table"` })
数据校验分为以下几个步骤:
Datasource
对应的数据库Schema
建表
这一步中,需要将Schema和Query中的表名加上uuid以避免在校验过程中遇到重名表(为了保证thread-safe
)ColumnType, 反射
等获取到对应Golang
基础类型的零值使用gofakeit
库mock
对应Golang
基础类型的值,构建SQL
后执行插入对应表;这里需要注意的是,在步骤3中获取到的「对应Golang
基础类型」可能并不能在mock
后再次插入数据库(由于数据库列类型和golang
类型并不是1:1的关系),所以我们需要针对诸如date, smallInt, id, jsonp
等类型编写mock
逻辑
SELECT * FROM <table_name> LIMIT 0并且部分类型需要限制值域范围,如time和smallInt等,以精确适配数据库类型的值域范围
Query
2, 3, 4, 5步骤均可能出现错误,3中的错误可忽略,其它步骤全部执行通过则确认该数据校验通过,同时收集返回结果以便做后续的面板展示
除了针对问题和schema
获取到执行的Query
,在这样的问题背景下,我们还希望了解得到这样的Query
的具体思路,以支持ABP调优中的【学习复杂函数问题】这一项,代替原来的基于llama_index
构建复杂函数文档的方案
首先我们可以直接从Query
中解析出全部用到的函数(在完成上述数据校验的步骤5校验后),并根据提前收集好的各数据源函数集合找到其对应功能(这样在利于LLM
推理分析的同时,生成的数据也将在训练时把各数据源的函数信息喂给LLM
,一举两得),以此为基础构建prompt
:
## InstructionShow me the steps to get the provided Query## ContextI've got a Query SQL based on a schema(CREATE TABLE SQL) and a specific task## Input Dataschema: "CREATE OR REPLACE TABLE hits(WatchID BIGINT, JavaEnable SMALLINT, Title TEXT, GoodEvent SMALLINT, EventTime TIMESTAMP, EventDate Date, CounterID INTEGER, ClientIP INTEGER, RegionID INTEGER, UserID BIGINT, CounterClass SMALLINT, OS SMALLINT, UserAgent SMALLINT, URL TEXT, Referer TEXT, IsRefresh SMALLINT, RefererCategoryID SMALLINT, RefererRegionID INTEGER, URLCategoryID SMALLINT, URLRegionID INTEGER, ResolutionWidth SMALLINT, ResolutionHeight SMALLINT, ResolutionDepth SMALLINT, FlashMajor SMALLINT, FlashMinor SMALLINT, FlashMinor2 TEXT, NetMajor SMALLINT, NetMinor SMALLINT, UserAgentMajor SMALLINT, UserAgentMinor VARCHAR(255), CookieEnable SMALLINT, JavascriptEnable SMALLINT, IsMobile SMALLINT, MobilePhone SMALLINT, MobilePhoneModel TEXT, Params TEXT, IPNetworkID INTEGER, TraficSourceID SMALLINT, SearchEngineID SMALLINT, SearchPhrase TEXT, AdvEngineID SMALLINT, IsArtifical SMALLINT, WindowClientWidth SMALLINT, WindowClientHeight SMALLINT, ClientTimeZone SMALLINT, ClientEventTime TIMESTAMP, SilverlightVersion1 SMALLINT, SilverlightVersion2 SMALLINT, SilverlightVersion3 INTEGER, SilverlightVersion4 SMALLINT, PageCharset TEXT, CodeVersion INTEGER, IsLink SMALLINT, IsDownload SMALLINT, IsNotBounce SMALLINT, FUniqID BIGINT, OriginalURL TEXT, HID INTEGER, IsOldCounter SMALLINT, IsEvent SMALLINT, IsParameter SMALLINT, DontCountHits SMALLINT, WithHash SMALLINT, HitColor CHAR, LocalEventTime TIMESTAMP, Age SMALLINT, Sex SMALLINT, Income SMALLINT, Interests SMALLINT, Robotness SMALLINT, RemoteIP INTEGER, WindowName INTEGER, OpenerName INTEGER, HistoryLength SMALLINT, BrowserLanguage TEXT, BrowserCountry TEXT, SocialNetwork TEXT, SocialAction TEXT, HTTPError SMALLINT, SendTiming INTEGER, DNSTiming INTEGER, ConnectTiming INTEGER, ResponseStartTiming INTEGER, ResponseEndTiming INTEGER, FetchTiming INTEGER, SocialSourceNetworkID SMALLINT, SocialSourcePage TEXT, ParamPrice BIGINT, ParamOrderID TEXT, ParamCurrency TEXT, ParamCurrencyID SMALLINT, OpenstatServiceName TEXT, OpenstatCampaignID TEXT, OpenstatAdID TEXT, OpenstatSourceID TEXT, UTMSource TEXT, UTMMedium TEXT, UTMCampaign TEXT, UTMContent TEXT, UTMTerm TEXT, FromTag TEXT, HasGCLID SMALLINT, RefererHash BIGINT, URLHash BIGINT, CLID INTEGER) ENGINE = MergeTree() ORDER BY (CounterID, EventDate, UserID, EventTime, WatchID)"question: "give me page view count bucketed by date and order by date between 2013-07-14 and 2013-07-15 where CounterID is 62 and is not refreshed"query: "SELECT DATE_TRUNC('day', EventTime) AS M, COUNT(*) AS PageViews FROM hits WHERE CounterID = 62 AND EventDate >= '2013-07-14' AND EventDate <= '2013-07-15' AND IsRefresh = 0 GROUP BY DATE_TRUNC('day', EventTime) ORDER BY DATE_TRUNC('day', EventTime)used_functions: [{ "func_name": "date_trunc", "feature": "Truncates a date to a specified precision"}, { "func_name": "count", "feature": "Counts the number of rows or not-NULL values"}]Solve by breaking the problem into steps. First, Analyze the question; second, understand the schema and bind the question; third, write the Query, and then explain why we used the functions## Output Indicator` ``` `## Step 1: Analyze the Question...## Step 2: Understand the Schema...## Step 3: Write the SQL Query...## Step 4: Explanation of Functions Used...` ``` `
值得注意的是,这里我使用了CoT
(Chain-of-Thought)的prompt
构建技巧,即Input Data
部分的最后一句话:
Solve by breaking the problem into steps. First, Analyze the question; second, understand the schema and bind the question; third, write the Query, and then explain why we used the functions
对于目前的LLM
来说,如今最困难的任务之一可能是需要某种形式的推理;而使用CoT
提示就可以通过中间推理步骤,让LLM
实现复杂的推理能力,有关CoT
的具体内容我会在另一篇Prompt-Engineering
的专题文章中介绍
最终该prompt可以实现这样的效果:
## Step 1: Analyze the Question
The question requires us to retrieve the page view count from a table called "hits" that has several columns. We need to filter the data by CounterID, date, and IsRefresh column. Finally, the query should group the data by the day and order it by the date.
## Step 2: Understand the Schema
The schema defines a table called "hits" with many columns that store information about the website visitors, their actions, and their devices. Each row represents a single event, and the table is sorted by CounterID, EventDate, UserID, EventTime, WatchID. The columns that are relevant to our query are:
- EventTime: the timestamp of the event
- CounterID: the ID of the counter that tracks the event
- IsRefresh: a flag that indicates whether the event was triggered by a page refresh
- EventDate: the date of the event
## Step 3: Write the SQL Query
Based on the above analysis, we can write the SQL query as follows:
` ``` `
SELECT DATE_TRUNC('day', EventTime) AS M, COUNT(*) AS PageViews
FROM hits
WHERE CounterID = 62
AND EventDate >= '2013-07-14'
AND EventDate <= '2013-07-15'
AND IsRefresh = 0
GROUP BY DATE_TRUNC('day', EventTime)
ORDER BY DATE_TRUNC('day', EventTime)
` ``` `
## Step 4: Explanation of Functions Used
We used the following functions in our query:
- `DATE_TRUNC`: This function truncates a timestamp to a specified time unit (in our case, day) and returns a new timestamp. We use it to group the data by day.
- `COUNT`: This function counts the number of rows in each group. We use it to count the number of page views in each day.
Both functions are common in SQL queries and are used to manipulate and summarize data.
有了这样的CoT
链prompt
,不仅能够让LLM
明白各个函数的具体功能,也能为LLM
提供text-to-sql
问题的推导思路,以实现一些复杂查询的推导;这一部分的分析也可以考虑移到数据校验之前,与数据生成的部分做一个整合,这样补充Analysis
信息后的数据集也经过了验证,最终模型在问题理解能力和SQL
推理能力都比未添加Analysis
信息前要好得多
实现了上述的基于GPT-3.5
生成数据集以及数据校验后,我们可以通过使用CI/CD pipeline
将数据集的构建和自动化数据校验过程结合起来,以确保我们能够快速、可靠地构建数据集并过滤掉低质量数据,同时也为后续上线使用中的反馈流程打好基础,确保我们的数据集始终处于最佳状态,并且可以随时部署到我们的模型中
基于prompt-collector
提供的比较友好的命令行调用方式,我们可以使用下述命令直接生成和校验AI-SQL的数据集:
# 可以分两步
prompt-collector generate -c 2000 -d clickhouse -o ${{ ci.workspace }}/generated/ --concurrent 8 --timeout 3h
prompt-collector validate --input ${{ ci.workspace }}/generated/ -o /data/prompt-collector/test/dataset/validated --concurrent 4
# 也支持生成数据并校验
prompt-collector generate -c 2000 -d clickhouse -o ${{ ci.workspace }}/generated/ --validate --concurrent 8 --timeout 3h
随后完成配套的流水线搭建,实现定时的数据集生成和数据集校验(这里将-c
设置到比较高的值并将--timeout
设置为与流水线触发周期间隔相同的时间,以实现24小时满载运行,并且timeout
参数保证了在遇到异常情况时能及时终止并将已生成的数据及时上传和持久化)
当前开启8并发(--concurrent 8
)的情况下,生产数据的速度可以达到6400条 / h
,单次请求消耗如下(生产5条数据),计算可得生产每条数据消耗约为0.0004美元
模型 | 问题消耗 Tokens | 回答消耗 Tokens | 总消耗 Tokens | 费用(美元) |
---|---|---|---|---|
gpt-3.5-turbo-0301 | 480 | 580 | 1060 | 0.00212 |
初期可以直接在完成生成和校验后将新数据写回仓库,直接开始运行训练任务
后期可以持续的写入数据库或是其它存储介质,方便做后续的处理和使用
我们可以选择直接使用开源数据集,如WikiSQL、SParC、HybridSQL、CoSQL
等的数据集,也可以使用基于GPT-3.5生成的数据集(在一些开源数据集匮乏的场景下吗,如clickhouse
复杂查询的text-to-sql
数据),在基于GPT-3.5生成数据集时也就需要使用prompt engineering
的技巧来提高生成数据集的效率和质量,并不断迭代更新结构化的prompt
,以支持和实现CoT
链等特性
在获取数据后,我们还需要进行严格的数据校验,以确保我们构建的text-to-sql
数据集是高质量的;除此之外,我们还需要进行SQL分析,以了解SQL的具体思路,将思路也融合进训练数据集中,提升其问题理解能力和SQL推理能力,以支持其构建更加复杂的查询
最后,我们可以使用CI实现自动化构建数据集,以持续支持我们的模型训练及反馈流程等,最终实现全自动炼丹的一整套框架
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。