[發明專利]決策樹模型的交互式建模方法、裝置、設備及存儲介質

有效

• 申請號: 201911024616.1
• 申請日: 2019-10-25
• 公開（公告）號: CN110782340B
• 公開（公告）日: 2021-09-07
• 發明（設計）人: 李詩琦;黃啟軍;唐興興;林冰垠;王躍
• 申請（專利權）人: 深圳前海微眾銀行股份有限公司
• 主分類號: G06Q40/02
• 分類號: G06Q40/02
• 代理公司: 深圳市世紀恒程知識產權代理事務所 44287
• 代理人: 王韜

本發明公開了一種決策樹模型的交互式建模方法、裝置、設備及存儲介質，所述方法包括：獲取待操作的決策樹模型和對決策樹模型進行操作的操作任務；根據操作任務對決策樹模型進行操作，并根據操作后的決策樹模型確定待計算的模型信息；將模型信息的計算任務分發給與本機設備連接的各分布式執行機，以供各分布式執行機根據各自的本地數據執行計算任務；根據各分布式執行機的計算結果得到模型信息，并可視化輸出操作后的決策樹模型和模型信息。本發明實現了聯合存儲有決策樹模型訓練數據的多個分布式執行機進行決策樹模型的交互式建模，從而解決了基于單機的交互式建模方式不能利用海量分布式存儲的訓練數據進行建模的問題。

技術領域

本發明涉及人工智能領域，尤其涉及一種決策樹模型的交互式建模方法、裝置、設備及存儲介質。

背景技術

由于決策樹模型具有較好的可解釋性，目前決策樹模型被廣泛應用于諸多領域，如金融機構會使用決策樹模型進行風險信貸建模。通常建模的方式并不是單純的使用機器學習訓練生成一顆決策樹，而是采用交互式建模型的方式，將決策樹模型的建模過程可視化，可以由人工設定節點拆分規則，使得決策樹按照預定的規則生長，建模人員通過反復進行交互操作和觀察效果，直到得出符合預期的決策樹模型。

目前常見的交互式建模工具是基于單機運行的，例如廣泛使用的SAS EM工具，單機運行限制了可處理的數據量大小，無法利用海量的分布式存儲的訓練數據來進行交互式建模。

發明內容

本發明的主要目的在于提供一種決策樹模型的交互式建模方法、裝置、設備及存儲介質，旨在解決目前基于單機的交互式建模方式無法利用海量分布式存儲的訓練數據來進行交互式建模的問題。

為實現上述目的，本發明提供一種決策樹模型的交互式建模方法，所述決策樹模型的交互式建模方法包括以下步驟：

獲取待操作的決策樹模型和對所述決策樹模型進行操作的操作任務；

根據所述操作任務對所述決策樹模型進行操作，并根據操作后的決策樹模型確定待計算的模型信息；

將所述模型信息的計算任務分發給與本機設備連接的各分布式執行機，以供各分布式執行機根據各自的本地數據執行所述計算任務；

根據各分布式執行機的計算結果得到所述模型信息，并可視化輸出操作后的決策樹模型和所述模型信息。

可選地，所述獲取待操作的決策樹模型和對所述決策樹模型進行操作的操作任務的步驟包括：

接收與本機設備連接的用戶端發送的交互操作指令；

根據所述交互操作指令從服務器獲取待操作的決策樹模型，以及根據所述交互操作指令確定對所述決策樹模型進行操作的操作任務；

根據各分布式執行機的計算結果得到所述模型信息的步驟之后，還包括：

將操作后的決策樹模型和所述模型信息緩存至所述服務器中；

所述可視化輸出操作后的決策樹模型和所述模型信息的步驟包括：

將操作后的決策樹模型和所述模型信息返回給所述用戶端，以供所述用戶端可視化輸出操作后的決策樹模型和所述模型信息。

可選地，當所述操作任務是對所述決策樹模型中目標節點的拆分操作、待計算的所述模型信息包括所述拆分操作中拆分規則的拆分增益時，

所述將所述模型信息的計算任務分發給與本機設備連接的各分布式執行機，以供各分布式執行機根據各自的本地數據執行所述計算任務的步驟包括：

將所述拆分操作中的拆分規則發送給各分布式執行機，以供各分布式執行機計算各自的本地數據在所述拆分規則下的統計信息；

所述根據各分布式執行機的計算結果得到所述模型信息的步驟包括：