[发明专利]数据预测模型的创建方法、装置、设备及存储介质

权利要求书

1.一种数据预测模型的创建方法，其特征在于，应用于包括第一设备和第二设备的联邦系统，所述第一设备为所述数据预测模型的发起方设备，所述第二设备为所述数据预测模型的参与方设备，所述第一设备用于创建所述数据预测模型，所述第二设备用于辅助所述第一设备创建所述数据预测模型，所述数据预测模型采用对称树结构，所述方法包括：

所述第一设备获取第一样本集合，所述第一样本集合为初始的对称树结构的第一层节点的待划分样本集合，所述第一样本集合包括第一设备和第二设备共有的样本；

所述第一设备根据所述第一设备确定的所述第一层节点的划分特征划分所述第一样本集合，得到所述对称树结构的第二层节点的样本集合，所述第二层节点包括两个节点；

所述第一设备根据所述第一设备确定的所述第二层节点的任意一个节点的划分特征，分别划分所述第二层节点的每个节点的样本集合，得到所述第二层节点的划分结果；

所述第一设备采用对所述第二层节点的样本集合的划分方式，分别划分所述对称树结构的所述第二层之下的每一层节点的样本集合，直至满足预设条件停止样本划分，将停止时的对称树结构作为所述数据预测模型。

2.根据权利要求1所述的方法，其特征在于，所述第一设备根据所述第一设备确定的所述第一层节点的划分特征划分所述第一样本集合，包括：

所述第一设备获取所述第一设备采用第一特征划分所述第一样本集合的第一样本纯度值，以及所述第二设备采用第二特征划分所述第一样本集合的第二样本纯度值；所述第一特征包括样本在所述第一设备的至少一个特征，所述第二特征包括样本在所述第二设备的至少一个特征；

所述第一设备根据所述第一样本纯度值和所述第二样本纯度值，确定所述第一层节点的划分特征；

所述第一设备根据所述第一层节点的划分特征划分所述第一样本集合。

3.根据权利要求2所述的方法，其特征在于，所述第一设备获取所述第一设备采用第一特征划分所述第一样本集合的第一样本纯度值，包括：

所述第一设备根据第一划分结果以及所述第一样本集合的标注结果，确定所述第一样本纯度值；所述第一划分结果是所述第一设备采用第一特征划分所述第一样本集合得到的。

4.根据权利要求2所述的方法，其特征在于，所述第一设备获取所述第二设备采用第二特征划分所述第一样本集合的第二样本纯度值，包括：

所述第一设备从所述第二设备获取第二划分结果，所述第二划分结果是所述第二设备采用第二特征划分所述第一样本集合得到的，所述第二划分结果用于所述第一设备确定所述第一层节点的划分特征；

所述第一设备根据所述第二划分结果以及所述第一样本集合的标注结果，确定所述第二样本纯度值。

5.根据权利要求2所述的方法，其特征在于，所述第一设备根据所述第一样本纯度值和所述第二样本纯度值，确定所述第一层节点的划分特征，包括：

所述第一设备将最大样本纯度值对应的划分特征作为所述第一层节点的划分特征。

6.根据权利要求1所述的方法，其特征在于，所述第二层之下的每一层节点包括第三层节点，所述第三层节点包括四个节点；所述第一设备根据所述第一设备对所述第二层节点的样本集合的划分方式，分别划分所述对称树结构的所述第二层之下的每一层节点的样本集合，包括：

所述第一设备根据所述第一设备确定的所述第三层节点的任意一个节点的划分特征，分别划分所述第三层节点的每个节点的样本集合。

7.根据权利要求1所述的方法，其特征在于，所述预设条件包括以下任意一项：

所有待划分节点的样本集合中样本数量小于或等于预设的最小样本数；

达到预设的对称树结构的最大层数；

达到预设的对称树结构的最大叶子节点数；

所有待划分节点的样本集合的样本纯度值大于预设阈值。

8.根据权利要求1-7任一项所述的方法，其特征在于，所述第二层节点的划分结果包括所述第二层节点的每个节点的子节点的样本集合，所述方法还包括：

所述第一设备向所述第二设备发送所述第二层节点的划分结果。