一种基于决策树的戒毒人员戒治效果评估方法与流程

技术特征：

1.一种基于决策树的戒毒人员戒治效果评估方法，其特征在于，包括以下步骤：

S1：目标函数选择：从戒毒人员多维数据中选择一个维度YD作为目标函数；

S2：特征选择：从戒毒人员多维数据选择一组特征FD；

S3：训练过程：根据目标函数YD和特征FD建立训练数据集TrainSet，训练决策树回归模型DTM，计算模型DTM中每个叶子节点的参数LNSTD和LNMEAN，保存决策树回归模型DTM、全体均值GMEAN、全体标准差GSTD、样本标准差LNSTD和样本均值LNMEAN，训练过程完成；

S4：评估过程；加载训练过程保存的决策树回归模型DTM、全体均值GMEAN、全体标准差GSTD、样本标准差LNSTD和样本均值LNMEAN，用决策树回归算法根据模型DTM预测被评估人员的目标函数YD值，获得命中的决策树回归模型DTM叶子节点计算LSS，根据被评估人员的目标函数YD值、全体均值GMEAN、全体标准差GSTD计算GSS，作为评估结果输出LSS和GSS。

2.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S3中，所述训练决策树回归模型DTM为从数据集TrainSet中，提取month等于mi的样本，放入子集ModelTrainSet用于训练决策树回归模型DTM，即提取第mi月的数据训练决策树回归模型DTM，mi取month的中间值或取mi＝12。

3.根据权利要求2所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S3中，所述子集ModelTrainSet训练决策树回归模型DTM过程中，控制叶子节点的最小样本数>MNS，其中10≤MNS<子集ModelTrainSet样本总数或叶子节点总数。

4.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S3中，所述计算决策树回归模型DTM中每个叶子节点的参数LNSTD和LNMEAN为将得到的决策树回归模型DTM中所有叶子节点放到统一的叶子节点数组lnodes中，叶子节点数为lnsize，lnsize等于lnodes的长度，计算命中叶子节点的所有数据集TrainSet样本的标准差LNSTD数组和均值LNMEAN数组，样本标准差LNSTD和样本均值LNMEAN均为二维数组，第一维表示月份，长度为36，第二维表示节点，长度为lnsize，LNSTD[m][i]的值为命中第i个叶子节点的第m个月样本的label的标准差，LNMEAN[m][i]的值为命中第i个叶子节点的第m个月样本的label的平均值。

5.根据权利要求4所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S3中，所述样本标准差LNSTD和样本均值LNMEAN的具体计算方法为：

S301：建立集合数组TSS，集合数组TSS为一个二维数组，第一维表示月份，长度为36，第二维表示节点，长度为叶子节点数lnsize，集合数组TSS的所有元素初始化为空集；

S302：列举数据集TrainSet集合中每一个样本x，用决策树回归算法预测x.features的预测值py，忽略预测值py，取预测过程中命中的决策树叶子节点在叶子节点数组lnodes中的下标lni，将样本x加入子集TSS[x.month][lni]；

S303：列举集合数组TSS的每个元素TSS[m][j]，TSS[m][j]为一个样本的子集，计算这个子集的元素label的均值和标准差，保存到LNSTD[m][i]和LNMEAN[m][i]中；

S304：建立一维集合数组GTSS，长度为36，所有元素初始化为空集，列举数据集TrainSet集合中每一个样本x，将x加入子集GTSS[x.month]；

S305：列举一维集合数组GTSS的每个元素GTSS[m]，GTSS[m]为一个样本的子集，计算这个子集所有样本的label的均值和标准差，保存到数组GMEAN[m]和GSTD[m]，GMEAN和GSTD为一维数组，表示全体的均值和标准差，下标m表示月份。

6.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S3中，所述数据集TrainSet为样本集合，每个样本对应多维度戒毒数据中一个人员的数据，每个样本有三个列：month、label和features，使用目标函数YD的值作为label，从多维度戒毒数据中提取选中特征FD的数据构造特征向量features，从多维度戒毒数据中提取戒毒时间作为mouth，以月为单位。

7.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S4中，所述具体评估过程为：

S401：从存储介质加载训练过程得到的决策树回归模型DTM、全体均值GMEAN、全体标准差GSTD、样本标准差LNSTD和样本均值LNMEAN；

S402：使用与数据集TrainSet样本features列相同的方法，提取被评估人员的特征向量features，用决策树回归算法根据决策树回归模型DTM预测features的目标函数YD属性值，忽略预测值，取得features命中决策树回归模型DTM中的叶子节点的下标lni，计算被评估人员的戒毒时间month，计算参数LSS＝(YD-LNMEAN[m][lni])/LNSTD[m][lni]；

S403：计算GSS＝(YD-GMEAN[m])/GSTD[m]；

S404：输出评估结果LSS和GSS，以及LSS和GSS指标随时间变化的趋势，作为被评估人员戒治效果指标YD的直观说明；

GSS＞0表示被评价人员的戒治效果优于整体平均水平，GSS＜0表示被评价人员的戒治效果比整体平均水平差；

当LSS＞0，表示被评估人员当前的戒治效果优于类似戒毒人员平均值，

-1＜LSS＜1，表示被评估人员戒治效果和类似戒毒人员的平均值偏差在1个标准差之内，标注其戒治效果为“正常”，

LSS＜-1表示戒毒人员戒治效果低于类似戒毒人员平均值超过1个标准差，标注其戒治效果为“差”，

LSS＞1表示戒毒人员戒治效果高于类似戒毒人员平均值超过1个标准差，标注其戒治效果为“优”；

当LSS和GSS的评估结果不同时，以LSS的评估结果为标准。

8.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S1中，所述目标函数YD为累计奖罚分、月度奖罚分、考试成绩、医疗检验结果和康复训练成绩中的任一种。

9.根据权利要求1所述的基于决策树的戒毒人员戒治效果评估方法，其特征在于，步骤S2中，所述特征FD为性别、年龄、吸食毒品种类和文化程度中的任一种。

技术总结
本发明提供一种基于决策树的戒毒人员戒治效果评估方法，属于机器学习技术领域。该方法包括以下技术步骤：S1：目标函数选择；S2：特征选择；S3：训练过程：S4：评估过程。本发明方法得到的戒毒人员戒治效果评估模型可以对戒毒人员进行定期的戒治效果评估，评估只需要从信息系统数据库中提取数据，评估过程简单成本低，无需加入额外的人为主观判断，准确率高，输出指标易于理解和把握，灵活性高，可以适应各地域不同的制度、不同的技术设备带来的巨大差异，适应性强，当制度变化，技术进步导致数据发生巨大变化之后，可以通过重新训练模型的方式快速适应变化。

技术研发人员：陆宇升;李家深;朱晓东;许金礼;陶炜;廖淑珍;
受保护的技术使用者：广西友迪资讯科技有限公司;
技术研发日：2021.07.29
技术公布日：2021.11.16