[发明专利]性别预测方法、装置、存储介质及电子设备

权利要求书

1.一种性别预测方法，其特征在于，包括：

获取已知性别用户使用电子设备的多维特征作为样本，并构建性别预测的样本集；

根据所述特征对于样本集分类的基尼指数信息增益对所述样本集进行分类，以构建出相应的分类回归树模型，所述分类回归树模型的输出包括男性、或者女性；

根据预测时间采集未知性别用户使用电子设备的多维特征作为预测样本；

根据所述预测样本和所述分类回归树模型预测所述未知性别用户的性别。

2.如权利要求1所述的性别预测方法，其特征在于，根据所述特征对于样本集的基尼指数信息增益对所述样本集进行划分，以构建出相应的分类回归树模型，包括：

生成分类回归树模型的根节点，并将所述样本集分配给所述根节点；

将所述根节点的样本集确定为当前待分类的目标样本集；

获取所述特征对于目标样本集分类的基尼指数信息增益；

根据所述基尼指数信息增益从所述特征中选取当前的划分特征及其对应的划分点；

根据所述划分特征和所述划分点对所述样本集进行划分，得到两个子样本集；

生成当前节点的子节点，并将所述去所述子样本集分配给相应的所述子节点；

判断所述子节点是否满足预设分类终止条件；

若否，则将所述目标样本集更新为所述子样本集，并返回执行获取所述特征对于目标样本集的基尼指数的步骤；

若是，则将所述子节点作为叶子节点，根据所述子样本集中样本的样本类别设置所述叶子节点的输出，所述样本类别包括男性、或者女性。

3.如权利要求2所述的性别预测方法，其特征在于，获取所述特征对于目标样本集分类的基尼指数信息增益，包括：

获取所述特征的取值对于目标样本集分类的基尼指数；

根据所述基尼指数，获取所述特征的取值对于目标样本集分类的基尼指数信息增益。

4.如权利要求3所述的性别预测方法，其特征在于，获取所述特征的取值对于目标样本集分类的基尼指数信息增益，包括：

根据所述特征的取值将所述目标样本集划分成第一子样本集和第二子样本集；

获取所述第一子样本集和所述第二子样本集中样本类别的概率；

根据所述样本类别的概率获取所述取值对于目标样本分类的基尼指数。

5.如权利要求3所述的性别预测方法，其特征在于，根据所述样本类别的概率获取所述取值对于目标样本分类的基尼指数，包括：

根据所述第一子样本集中样本类别的概率获取所述特征为取值时对于目标样本集分类的第一基尼指数；

根据所述第二子样本集中样本类别的概率获取所述特征不为取值时对于目标样本集分类的第二基尼指数；

根据所述基尼指数，获取所述特征对于目标样本集分类的基尼指数信息增益，包括：

根据所述第一基尼指数、第一子样本集与目标样本集的样本数量比值、第二基尼指数、以及第二子样本集与目标样本集的样本数量比值，获取所述特征的取值对于目标样本集分类的基尼指数信息增益。

6.如权利要求5所述的性别预测方法，其特征在于，根据所述基尼指数，获取所述特征对于目标样本集分类的基尼指数信息增益，包括：

通过如下公式计算出特征对于目标样本集分类的基尼指数信息增益：

其中，Gini(D,A)为特征A对于目标样本集D分类的基尼指数信息增益，Gini(D₁)为特征A为取值a时对于目标样本D分类的基尼指数，Gini(D₂)为A不为取值a时对于目标样本D分类的基尼指数；a为特征A的一种取值，D1和D2为基于特征A＝a对目标样本集D划分后得到的两个子样本集。

7.如权利要求2所述的性别预测方法，其特征在于，根据所述基尼指数信息增益从所述特征中选取当前的划分特征及其对应的划分点，包括：

从所述基尼指数信息增益中确定最小的目标基尼指数信息增益；

将所述目标基尼指数信息增益的特征及其取值，分别作为划分特征和划分点。