[發明專利]一種模型訓練的方法及裝置

在審

• 申請號: 202110690782.6
• 申請日: 2021-06-22
• 公開（公告）號: CN113344201A
• 公開（公告）日: 2021-09-03
• 發明（設計）人: 馮家琪;肖垚;黃培浩;周翔;陳勝
• 申請（專利權）人: 北京三快在線科技有限公司
• 主分類號: G06N3/08
• 分類號: G06N3/08;G06N3/04;G06F16/953;G06F16/9538
• 代理公司: 北京曼威知識產權代理有限公司 11709
• 代理人: 方志煒

本說明書公開了一種模型訓練的方法及裝置，可以獲取訓練樣本，該訓練樣本中包含搜索語句，相應的搜索結果、該搜索語句對應的第一標注信息以及該搜索結果對應的第二標注信息。而后，將該搜索結果以及該搜索語句輸入到待訓練的業務模型中，得到該搜索結果在各業務下的排序評分，以及該搜索語句與各業務之間的匹配度，針對每個業務，根據該搜索語句與該業務之間匹配度，補償該搜索結果在該業務下的排序評分，得到該搜索結果在該業務下的補償后排序評分，根據該搜索結果在各業務下的補償后排序評分、該搜索語句與各業務之間的匹配度、第一標注信息以及第二標注信息，訓練業務模型，在降低成本的同時提高了排序的準確性。

技術領域

本說明書涉及計算機技術領域，尤其涉及一種模型訓練的方法及裝置。

背景技術

隨著信息技術的不斷發展，一個業務平臺可以為用戶提供各種各樣的業務，而為了便于用戶快速找到所需執行的業務，業務平臺可以為用戶提供搜索的功能，用戶輸入一定的搜索語句，業務平臺可以為用戶查詢到相應的結果展示給用戶。

在實際應用中，業務平臺可以提供給用戶一個統一的搜索入口，用戶可以通過該搜索入口搜索到業務平臺中任意一項業務中的搜索結果，例如，在業務平臺中，存在訂酒店、預約餐廳、點外賣等業務，用戶可以在統一搜索入口中搜索某個酒店、某個餐廳等。

在現有技術中，業務平臺可以針對每種業務構建一種排序模型，該排序模型負責對在這一業務中與用戶的搜索語句相關的搜索結果進行排序，但是這種方式業務平臺需要維護多個模型，成本較高。

所以，如何在保證對搜索結果排序的準確性的條件下降低維護排序模型的成本，則是一個亟待解決的問題。

發明內容

本說明書提供一種模型訓練的方法及裝置，以部分的解決現有技術存在的上述問題。

本說明書采用下述技術方案：

本說明書提供了一種模型訓練的方法，包括：

獲取訓練樣本，所述訓練樣本中包含搜索語句，所述搜索語句對應的搜索結果、所述搜索語句對應的第一標注信息以及所述搜索結果對應的第二標注信息，所述第一標注信息用于表示與所述搜索語句相關的業務，所述第二標注信息用于表示用戶針對所述搜索結果的實際操作結果；

將所述搜索結果以及所述搜索語句輸入到待訓練的業務模型中，得到所述搜索結果在各業務下的排序評分，以及所述搜索語句與各業務之間的匹配度；

針對每個業務，根據所述搜索語句與該業務之間匹配度，補償所述搜索結果在該業務下的排序評分，得到所述搜索結果在該業務下的補償后排序評分；

根據所述搜索結果在各業務下的補償后排序評分、所述搜索語句與各業務之間的匹配度、所述第一標注信息以及所述第二標注信息，訓練所述業務模型。

可選地，所述業務模型中包含排序模型以及業務匹配模型；

將所述搜索結果以及所述搜索語句輸入到待訓練的業務模型中，得到所述搜索結果在各業務下的排序評分，以及所述搜索語句與各業務之間的匹配度，具體包括：

將所述搜索結果輸入到所述排序模型中，得到所述搜索結果在各業務下的排序評分，以及將所述搜索語句輸入到所述業務匹配模型中，得到所述搜索語句與各業務之間的匹配度。

可選地，所述排序模型中包含特征提取層以及各業務對應的業務層；

將所述搜索結果輸入到所述排序模型中，得到所述搜索結果在各業務下的排序評分，具體包括：

將所述搜索結果輸入到所述特征提取層中，得到所述搜索結果對應的特征向量；

針對每個業務，將所述特征向量輸入到該業務對應的業務層中，得到所述搜索結果在該業務下的排序評分。