[發明專利]一種房間分類模型構建方法及裝置、房間分類方法及裝置

在審

• 申請號: 202110139999.8
• 申請日: 2021-02-01
• 公開（公告）號: CN112861692A
• 公開（公告）日: 2021-05-28
• 發明（設計）人: 文琦;王偉;戴坤龍;李文生
• 申請（專利權）人: 電子科技大學中山學院
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 余菲

1.一種房間分類模型構建方法，其特征在于，包括：

獲取訓練樣本集，所述訓練樣本集包括多個房間數據樣本，每一房間數據樣本包括房間的多個屬性、每一屬性對應的數據類型以及房間類型，所述房間的多個屬性包括房間的幾何信息和門窗信息；

根據所述訓練樣本集中每一房間數據樣本的多個屬性、每一屬性對應的數據類型以及房間類型構建決策樹，以獲得構建完成的房間分類模型。

2.根據權利要求1所述的方法，其特征在于，所述根據所述訓練樣本集中每一房間數據樣本的多個屬性、每一屬性對應的數據類型以及房間類型構建決策樹，包括：

根據所述訓練樣本集中每一房間數據樣本的多個屬性和房間類型計算每一屬性對應的信息增益率；

將信息增益率最大的屬性作為根節點；

根據所述根節點對應的數據類型對所述根節點進行分支劃分，以獲得所述根節點的每一分支；

根據每一分支對應的房間數據樣本生成每一分支對應的更新訓練樣本集；

根據每一分支對應的更新訓練樣本集確定每一分支對應的根節點，直至每一分支對應的房間數據樣本中屬于同一房間類型的樣本比例滿足預設比例，以完成決策樹的構建。

3.根據權利要求2所述的方法，其特征在于，所述數據類型包括離散型和連續型，所述根據所述根節點對應的數據類型對所述根節點進行分支劃分，包括：

判斷所述根節點對應的數據類型是否為連續型；

若是，則對所述根節點對應屬性的多個房間數據樣本的數據進行依次排序，以獲得數據排序集合；

根據所述數據排序集合確定所述數據排序集合對應的分割點；

根據所述分割點將所述數據排序集合劃分為第一集合和第二集合；

判斷分割點的個數是否達到預設個數；

若是，則執行根據每一分支對應的房間數據樣本生成每一分支對應的更新訓練樣本集的步驟；

若否，則根據所述第一集合和第二集合確定待分割集合，并對所述待分割集合進行分割，直至分割點達到預設個數，以獲得所述根節點的每一分支。

4.根據權利要求3所述的方法，其特征在于，所述數據排序集合為按照多個房間數據樣本的數據的數值由小到大進行排序，所述根據所述數據排序集合確定所述數據排序集合對應的分割點，包括：

記錄所述數據排序集合中數值最大的樣本對應的房間類型；

按照數值由大到小依次遍歷所述數據排序集中的每一樣本對應的房間類型，判斷遍歷的樣本對應的房間類型與所述數值最大的樣本對應的房間類型是否一致；

在遍歷的樣本對應的房間類型與所述數值最大的樣本對應的房間類型不一致時，獲取不一致時遍歷的樣本對應的數值；

根據所述不一致時遍歷的樣本對應的數值確定所述數據排序集合對應的分割點。

5.根據權利要求4所述的方法，其特征在于，所述根據所述不一致時遍歷的樣本對應的數值確定所述數據排序集合對應的分割點，包括：

根據不一致時遍歷的樣本對應的數值V_f確定所述數據排序集合對應的分割點T_j，其中，公式為：

6.根據權利要求4所述的方法，其特征在于，所述根據所述第一集合和第二集合確定待分割集合，并對所述待分割集合進行分割，直至分割點達到預設個數，以獲得所述根節點的每一分支，包括：

計算所述第一集合和第二集合分別對應的信息熵；

選擇信息熵最大的集合作為所述待分割集合；

根據所述待分割集合確定所述待分割集合對應的分割點；

根據所述待分割集合對應的分割點對所述待分割集合進行分割并返回所述判斷分割點的個數是否達到預設個數的步驟。