[發(fā)明專利]決策樹的構(gòu)建方法和裝置、存儲(chǔ)介質(zhì)及電子裝置

在審

• 申請?zhí)枺?/td>: 202210225899.1
• 申請日: 2022-03-07
• 公開（公告）號(hào): CN114662574A
• 公開（公告）日: 2022-06-24
• 發(fā)明（設(shè)計(jì)）人: 杜心達(dá)
• 申請（專利權(quán)）人: 興業(yè)消費(fèi)金融股份公司
• 主分類號(hào): G06K9/62
• 分類號(hào): G06K9/62;G06N5/00;G06N20/00
• 代理公司: 北京康信知識(shí)產(chǎn)權(quán)代理有限責(zé)任公司 11240
• 代理人: 劉曉燕

1.一種決策樹的構(gòu)建方法，其特征在于，包括：

獲取多個(gè)超參數(shù)字典，以及獲取訓(xùn)練樣本集，其中，所述多個(gè)超參數(shù)字典中的每個(gè)超參數(shù)字典包含一組預(yù)設(shè)超參數(shù)和一個(gè)隨機(jī)狀態(tài)值，所述訓(xùn)練樣本集中的每個(gè)訓(xùn)練樣本包含與一個(gè)訓(xùn)練對象對應(yīng)的一組屬性的屬性值；

使用所述多個(gè)超參數(shù)字典中的每個(gè)超參數(shù)字典和所述訓(xùn)練樣本集分別進(jìn)行決策樹構(gòu)建，得到多個(gè)決策樹，其中，在進(jìn)行決策樹構(gòu)建的過程中僅做一次迭代，所述多個(gè)決策樹中的每個(gè)決策樹包含的一組節(jié)點(diǎn)中，每個(gè)非葉子節(jié)點(diǎn)與所述一組屬性中的一個(gè)屬性對應(yīng)；

從所述多個(gè)決策樹中篩選出至少一個(gè)候選決策樹，其中，所述至少一個(gè)候選決策樹中的每個(gè)候選決策樹包含的一組節(jié)點(diǎn)滿足目標(biāo)節(jié)點(diǎn)條件；

根據(jù)所述每個(gè)候選決策樹的目標(biāo)評估參數(shù)，從所述至少一個(gè)候選決策樹選取目標(biāo)決策樹。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述獲取多個(gè)超參數(shù)字典，包括：

生成隨機(jī)狀態(tài)值隊(duì)列，其中，所述隨機(jī)狀態(tài)值隊(duì)列包含互不相同的多個(gè)隨機(jī)狀態(tài)值；

從所述隨機(jī)狀態(tài)值隊(duì)列中分別選取隨機(jī)狀態(tài)值與所述一組預(yù)設(shè)超參數(shù)進(jìn)行組合，得到所述多個(gè)超參數(shù)字典。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述使用所述多個(gè)超參數(shù)字典中的每個(gè)超參數(shù)字典和所述訓(xùn)練樣本集分別進(jìn)行決策樹構(gòu)建，得到多個(gè)決策樹，包括：

使用所述多個(gè)超參數(shù)字典中的每個(gè)超參數(shù)字典分別作為決策樹的超參數(shù)字典進(jìn)行決策樹構(gòu)建，得到多個(gè)初始決策樹；

使用所述訓(xùn)練樣本集分別對所述多個(gè)初始決策樹中的每個(gè)初始決策樹進(jìn)行訓(xùn)練，得到所述多個(gè)決策樹，其中，在使用所述訓(xùn)練樣本集對所述每個(gè)初始決策樹進(jìn)行訓(xùn)練的過程中僅做一次迭代，得到一個(gè)決策樹。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述從所述多個(gè)決策樹中篩選出至少一個(gè)候選決策樹，包括：

根據(jù)第一樣本集中的每個(gè)第一樣本包含的所述一組屬性的屬性值，確定所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)下的第一樣本；

根據(jù)第二樣本集中的每個(gè)第二樣本包含的所述一組屬性的屬性值，確定所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)下的第二樣本；

根據(jù)與所述每個(gè)決策樹對應(yīng)的第一占比和與所述每個(gè)決策樹對應(yīng)的第二占比，從所述多個(gè)決策樹中篩選出所述至少一個(gè)候選決策樹，其中，與所述每個(gè)決策樹對應(yīng)的第一占比為所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)下的第一樣本中目標(biāo)樣本的占比，與所述每個(gè)決策樹對應(yīng)的第二占比為所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)下的第二樣本中所述目標(biāo)樣本的占比，所述目標(biāo)樣本為具有目標(biāo)標(biāo)簽的樣本，與所述每個(gè)候選決策樹對應(yīng)的第一占比和與所述每個(gè)候選決策樹對應(yīng)的第二占比一致。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，所述根據(jù)與所述每個(gè)決策樹對應(yīng)的第一占比和與所述每個(gè)決策樹對應(yīng)的第二占比，從所述多個(gè)決策樹中篩選出所述至少一個(gè)候選決策樹，包括：

按照與所述每個(gè)決策樹對應(yīng)的第一占比對所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)進(jìn)行排序，得到與所述每個(gè)決策樹對應(yīng)的第一排序結(jié)果；

按照與所述每個(gè)決策樹對應(yīng)的第二占比對所述每個(gè)決策樹的每個(gè)葉子節(jié)點(diǎn)進(jìn)行排序，得到與所述每個(gè)決策樹對應(yīng)的第二排序結(jié)果；

從所述多個(gè)決策樹中篩選出對應(yīng)的第一排序結(jié)果和對應(yīng)的第二排序結(jié)果相同的決策樹，得到所述至少一個(gè)候選決策樹。