本申請(qǐng)涉及計(jì)算機(jī)，尤其涉及一種訓(xùn)練代碼生成方法、裝置、終端和計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)。

背景技術(shù)：

1、隨著人工智能的快速發(fā)展，代碼生成大模型逐步成為代碼生成的重要工具。代碼生成大模型能夠根據(jù)輸入的提示自動(dòng)生成相應(yīng)的代碼，從而幫助開(kāi)發(fā)者提高開(kāi)發(fā)效率和代碼質(zhì)量。然而，代碼生成大模型的訓(xùn)練需要大量質(zhì)量較高的代碼數(shù)據(jù)，常規(guī)的代碼數(shù)據(jù)往往質(zhì)量參差不齊，限制了代碼生成大模型的代碼生成效果。

技術(shù)實(shí)現(xiàn)思路

1、本申請(qǐng)實(shí)施例提供一種訓(xùn)練代碼生成方法、裝置、終端和計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，可以解決代碼生成大模型的訓(xùn)練需要大量質(zhì)量較高的代碼數(shù)據(jù)，常規(guī)的代碼數(shù)據(jù)往往質(zhì)量參差不齊，限制了代碼生成大模型的代碼生成效果的技術(shù)問(wèn)題。

2、第一方面，本申請(qǐng)實(shí)施例提供一種訓(xùn)練代碼生成方法，所述方法包括：

3、對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，確定所述目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段；

4、針對(duì)所述關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，基于所述關(guān)鍵代碼片段和所述目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞；

5、通過(guò)目標(biāo)代碼補(bǔ)全模型生成針對(duì)所述訓(xùn)練代碼生成提示詞的至少兩個(gè)參考訓(xùn)練代碼，針對(duì)所述目標(biāo)任務(wù)運(yùn)行所述參考訓(xùn)練代碼以從所述參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼。

6、可選地，所述對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，包括：

7、從目標(biāo)代碼數(shù)據(jù)庫(kù)中確定源代碼，對(duì)所述源代碼進(jìn)行類(lèi)型檢查處理得到類(lèi)型檢查結(jié)果，利用所述類(lèi)型檢查結(jié)果對(duì)所述源代碼進(jìn)行代碼篩選得到初始代碼；

8、對(duì)所述初始代碼進(jìn)行代碼完整性檢測(cè)處理得到完整性檢測(cè)結(jié)果，將所述完整性檢測(cè)結(jié)果為代碼完整類(lèi)型的初始代碼作為完整代碼，對(duì)所述完整代碼進(jìn)行代碼去重處理得到目標(biāo)完整代碼。

9、可選地，所述關(guān)鍵代碼片段具有可復(fù)用性；所述確定所述目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段，包括：

10、對(duì)所述目標(biāo)完整代碼進(jìn)行業(yè)務(wù)邏輯代碼片段提取處理，得到業(yè)務(wù)邏輯代碼片段，將所述業(yè)務(wù)邏輯代碼片段作為關(guān)鍵代碼片段；和/或，

11、對(duì)所述目標(biāo)完整代碼進(jìn)行決策代碼片段提取處理，得到?jīng)Q策代碼片段，將所述決策代碼片段作為關(guān)鍵代碼片段。

12、可選地，所述針對(duì)所述關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，包括：

13、獲取預(yù)先建立的任務(wù)數(shù)據(jù)庫(kù)，所述任務(wù)數(shù)據(jù)庫(kù)包括參考任務(wù)以及與所述參考任務(wù)關(guān)聯(lián)的至少一種任務(wù)執(zhí)行邏輯特征，所述任務(wù)執(zhí)行邏輯特征包括多個(gè)參考邏輯特征；

14、確定所述關(guān)鍵代碼片段對(duì)應(yīng)的代碼邏輯特征，利用所述代碼邏輯特征對(duì)所述任務(wù)數(shù)據(jù)庫(kù)中的參考邏輯特征進(jìn)行特征匹配處理，得到所述代碼邏輯特征匹配的目標(biāo)邏輯特征；

15、從所述任務(wù)執(zhí)行邏輯特征中確定所述目標(biāo)邏輯特征對(duì)應(yīng)的至少一個(gè)目標(biāo)任務(wù)執(zhí)行邏輯特征，將所述至少一個(gè)目標(biāo)任務(wù)執(zhí)行邏輯特征對(duì)應(yīng)的參考任務(wù)作為目標(biāo)任務(wù)。

16、可選地，所述基于所述關(guān)鍵代碼片段和所述目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞，包括：

17、確定所述目標(biāo)任務(wù)的任務(wù)內(nèi)容，基于所述任務(wù)內(nèi)容生成所述目標(biāo)任務(wù)對(duì)應(yīng)的任務(wù)描述；

18、對(duì)所述任務(wù)描述和所述關(guān)鍵代碼片段進(jìn)行描述關(guān)聯(lián)處理，得到目標(biāo)任務(wù)描述；

19、獲取訓(xùn)練代碼生成提示詞模板，基于所述目標(biāo)任務(wù)描述對(duì)所述訓(xùn)練代碼生成提示詞模板進(jìn)行描述編輯處理，得到訓(xùn)練代碼生成提示詞。

20、可選地，所述針對(duì)所述目標(biāo)任務(wù)運(yùn)行所述參考訓(xùn)練代碼以從所述參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼，包括：

21、針對(duì)所述目標(biāo)任務(wù)運(yùn)行所述參考訓(xùn)練代碼確定代碼運(yùn)行信息，基于所述代碼運(yùn)行信息確定所述參考訓(xùn)練代碼針對(duì)所述目標(biāo)任務(wù)的任務(wù)完成度；

22、利用所述任務(wù)完成度從所述參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼。

23、可選地，所述方法還包括：

24、獲取多個(gè)目標(biāo)任務(wù)對(duì)應(yīng)的目標(biāo)訓(xùn)練代碼，通過(guò)所述目標(biāo)任務(wù)確定所述目標(biāo)訓(xùn)練代碼對(duì)應(yīng)的任務(wù)完成度樣本標(biāo)簽；

25、控制所述初始代碼生成大模型生成所述目標(biāo)任務(wù)對(duì)應(yīng)的參考代碼，通過(guò)所述目標(biāo)任務(wù)確定所述參考代碼的任務(wù)完成度參考標(biāo)簽；

26、利用相同目標(biāo)任務(wù)對(duì)應(yīng)的任務(wù)完成度樣本標(biāo)簽和任務(wù)完成度參考標(biāo)簽確定模型損失值，采用所述模型損失值對(duì)所述初始代碼生成大模型進(jìn)行模型參數(shù)調(diào)整，直至所述初始代碼生成大模型完成模型訓(xùn)練得到目標(biāo)代碼生成大模型。

27、第二方面，本申請(qǐng)實(shí)施例提供一種訓(xùn)練代碼生成裝置，所述裝置包括：

28、生成模塊，適于對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，確定所述目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段；

29、構(gòu)造模塊，適于針對(duì)所述關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，基于所述關(guān)鍵代碼片段和所述目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞；

30、訓(xùn)練代碼生成模塊，適于通過(guò)目標(biāo)代碼補(bǔ)全模型生成針對(duì)所述訓(xùn)練代碼生成提示詞的至少兩個(gè)參考訓(xùn)練代碼，針對(duì)所述目標(biāo)任務(wù)運(yùn)行所述參考訓(xùn)練代碼以從所述參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼。

31、第三方面，本申請(qǐng)實(shí)施例提供一種終端，該終端包括：

32、處理器；以及

33、被安排成存儲(chǔ)計(jì)算機(jī)可執(zhí)行指令的存儲(chǔ)器，所述可執(zhí)行指令在被執(zhí)行時(shí)使所述處理器執(zhí)行根據(jù)上述任一項(xiàng)所述的方法。

34、第四方面，本申請(qǐng)實(shí)施例提供一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其中，所述計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)一個(gè)或多個(gè)程序，所述一個(gè)或多個(gè)程序當(dāng)被處理器執(zhí)行時(shí)，實(shí)現(xiàn)上述中任一項(xiàng)所述的方法。

35、本申請(qǐng)實(shí)施例提供的技術(shù)方案帶來(lái)的有益效果至少包括：通過(guò)對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，從而確保目標(biāo)完整代碼的代碼質(zhì)量。同時(shí)，由于目標(biāo)完整代碼可以獨(dú)立運(yùn)行而無(wú)需額外的代碼，使得目標(biāo)完整代碼可以獨(dú)立的實(shí)現(xiàn)至少一項(xiàng)功能，進(jìn)而使目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段可以在至少一項(xiàng)功能對(duì)應(yīng)的功能代碼中發(fā)揮作用，即關(guān)鍵代碼片段存在真實(shí)的應(yīng)用場(chǎng)景。由于目標(biāo)完整代碼的代碼質(zhì)量得到有效保證，使得目標(biāo)完整代碼中的關(guān)鍵代碼片段的代碼質(zhì)量同樣也得到了保證。

36、之后，針對(duì)關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，從而確定關(guān)鍵代碼片段可以應(yīng)用的一個(gè)或多個(gè)目標(biāo)任務(wù)，使得可以對(duì)關(guān)鍵代碼片段的應(yīng)用場(chǎng)景進(jìn)行發(fā)散拓展。然后，利用關(guān)鍵代碼片段和目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞，從而控制目標(biāo)代碼補(bǔ)全模型生成至少兩個(gè)參考訓(xùn)練代碼。為了確保最后得到的目標(biāo)訓(xùn)練代碼的代碼質(zhì)量，針對(duì)目標(biāo)任務(wù)運(yùn)行參考訓(xùn)練代碼，通過(guò)參考訓(xùn)練代碼的運(yùn)行表現(xiàn)來(lái)篩選出目標(biāo)訓(xùn)練代碼，從而使得利用每個(gè)關(guān)鍵代碼片段可以拓展出至少一個(gè)目標(biāo)任務(wù)對(duì)應(yīng)的高質(zhì)量目標(biāo)訓(xùn)練代碼，進(jìn)而解決了代碼生成大模型的訓(xùn)練需要大量質(zhì)量較高的代碼數(shù)據(jù)，常規(guī)的代碼數(shù)據(jù)往往質(zhì)量參差不齊，限制了代碼生成大模型的代碼生成效果的技術(shù)問(wèn)題。

技術(shù)特征：

1.一種訓(xùn)練代碼生成方法，其中，所述方法包括：

2.根據(jù)權(quán)利要求1所述的方法，其中，所述對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，包括：

3.根據(jù)權(quán)利要求1或2所述的方法，其中，所述關(guān)鍵代碼片段具有可復(fù)用性；所述確定所述目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段，包括：

4.根據(jù)權(quán)利要求1所述的方法，其中，所述針對(duì)所述關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，包括：

5.根據(jù)權(quán)利要求1所述的方法，其中，所述基于所述關(guān)鍵代碼片段和所述目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞，包括：

6.根據(jù)權(quán)利要求1所述的方法，其中，所述針對(duì)所述目標(biāo)任務(wù)運(yùn)行所述參考訓(xùn)練代碼以從所述參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼，包括：

7.根據(jù)權(quán)利要求6所述的方法，其中，所述方法還包括：

8.一種訓(xùn)練代碼生成裝置，其中，所述裝置包括：

9.一種終端，其中，該終端包括：

10.一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其中，所述計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)一個(gè)或多個(gè)程序，所述一個(gè)或多個(gè)程序當(dāng)被處理器執(zhí)行時(shí)，實(shí)現(xiàn)權(quán)利要求1至7中任一項(xiàng)所述的方法。

技術(shù)總結(jié)
本申請(qǐng)涉及計(jì)算機(jī)技術(shù)領(lǐng)域，本申請(qǐng)公開(kāi)了一種訓(xùn)練代碼生成方法、裝置、終端和計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，該方法包括：對(duì)目標(biāo)代碼數(shù)據(jù)庫(kù)進(jìn)行代碼篩選處理得到目標(biāo)完整代碼，確定目標(biāo)完整代碼對(duì)應(yīng)的關(guān)鍵代碼片段；之后針對(duì)關(guān)鍵代碼片段進(jìn)行任務(wù)預(yù)測(cè)處理得到至少一個(gè)目標(biāo)任務(wù)，基于關(guān)鍵代碼片段和目標(biāo)任務(wù)構(gòu)造訓(xùn)練代碼生成提示詞；最后，通過(guò)目標(biāo)代碼補(bǔ)全模型生成針對(duì)訓(xùn)練代碼生成提示詞的至少兩個(gè)參考訓(xùn)練代碼，針對(duì)目標(biāo)任務(wù)運(yùn)行參考訓(xùn)練代碼以從參考訓(xùn)練代碼中確定目標(biāo)訓(xùn)練代碼。從而解決代碼生成大模型的訓(xùn)練需要大量質(zhì)量較高的代碼數(shù)據(jù)，常規(guī)的代碼數(shù)據(jù)往往質(zhì)量參差不齊，限制了代碼生成大模型的代碼生成效果的技術(shù)問(wèn)題。

技術(shù)研發(fā)人員：梁博,楊亦威,周航宇
受保護(hù)的技術(shù)使用者：北京奇虎科技有限公司
技術(shù)研發(fā)日：
技術(shù)公布日：2025/1/6