【技术实现步骤摘要】
决策树模型的交互式建模方法、装置、设备及存储介质
本专利技术涉及人工智能领域,尤其涉及一种决策树模型的交互式建模方法、装置、设备及存储介质。
技术介绍
由于决策树模型具有较好的可解释性,目前决策树模型被广泛应用于诸多领域,如金融机构会使用决策树模型进行风险信贷建模。通常建模的方式并不是单纯的使用机器学习训练生成一颗决策树,而是采用交互式建模型的方式,将决策树模型的建模过程可视化,可以由人工设定节点拆分规则,使得决策树按照预定的规则生长,建模人员通过反复进行交互操作和观察效果,直到得出符合预期的决策树模型。目前常见的交互式建模工具是基于单机运行的,例如广泛使用的SASEM工具,单机运行限制了可处理的数据量大小,无法利用海量的分布式存储的训练数据来进行交互式建模。
技术实现思路
本专利技术的主要目的在于提供一种决策树模型的交互式建模方法、装置、设备及存储介质,旨在解决目前基于单机的交互式建模方式无法利用海量分布式存储的训练数据来进行交互式建模的问题。为实现上述目的,本专利技术提供一种决策树模型的交互式建模方法,所述决策树模型的交互式建模方法包括以下步骤:获取待操作的决策树模型和对所述决策树模型进行操作的操作任务;根据所述操作任务对所述决策树模型进行操作,并根据操作后的决策树模型确定待计算的模型信息;将所述模型信息的计算任务分发给与本机设备连接的各分布式执行机,以供各分布式执行机根据各自的本地数据执行所述计算任务;根据各分布式执行机的计算结果得到所述模型信 ...
【技术保护点】
1.一种决策树模型的交互式建模方法,其特征在于,所述决策树模型的交互式建模方法包括以下步骤:/n获取待操作的决策树模型和对所述决策树模型进行操作的操作任务;/n根据所述操作任务对所述决策树模型进行操作,并根据操作后的决策树模型确定待计算的模型信息;/n将所述模型信息的计算任务分发给与本机设备连接的各分布式执行机,以供各分布式执行机根据各自的本地数据执行所述计算任务;/n根据各分布式执行机的计算结果得到所述模型信息,并可视化输出操作后的决策树模型和所述模型信息。/n
【技术特征摘要】
1.一种决策树模型的交互式建模方法,其特征在于,所述决策树模型的交互式建模方法包括以下步骤:
获取待操作的决策树模型和对所述决策树模型进行操作的操作任务;
根据所述操作任务对所述决策树模型进行操作,并根据操作后的决策树模型确定待计算的模型信息;
将所述模型信息的计算任务分发给与本机设备连接的各分布式执行机,以供各分布式执行机根据各自的本地数据执行所述计算任务;
根据各分布式执行机的计算结果得到所述模型信息,并可视化输出操作后的决策树模型和所述模型信息。
2.如权利要求1所述的决策树模型的交互式建模方法,其特征在于,所述获取待操作的决策树模型和对所述决策树模型进行操作的操作任务的步骤包括:
接收与本机设备连接的用户端发送的交互操作指令;
根据所述交互操作指令从服务器获取待操作的决策树模型,以及根据所述交互操作指令确定对所述决策树模型进行操作的操作任务;
所述根据各分布式执行机的计算结果得到所述模型信息的步骤之后,还包括:
将操作后的决策树模型和所述模型信息缓存至所述服务器中;
所述可视化输出操作后的决策树模型和所述模型信息的步骤包括:
将操作后的决策树模型和所述模型信息返回给所述用户端,以供所述用户端可视化输出操作后的决策树模型和所述模型信息。
3.如权利要求1或2任一项所述的决策树模型的交互式建模方法,其特征在于,当所述操作任务是对所述决策树模型中目标节点的拆分操作、待计算的所述模型信息包括所述拆分操作中拆分规则的拆分增益时,
所述将所述模型信息的计算任务分发给与本机设备连接的各分布式执行机,以供各分布式执行机根据各自的本地数据执行所述计算任务的步骤包括:
将所述拆分操作中的拆分规则发送给各分布式执行机,以供各分布式执行机计算各自的本地数据在所述拆分规则下的统计信息;
所述根据各分布式执行机的计算结果得到所述模型信息的步骤包括:
根据各分布式执行机计算得到的所述统计信息计算所述拆分规则的拆分增益。
4.如权利要求3所述的决策树模型的交互式建模方法,其特征在于,当待计算的所述模型信息还包括所述目标节点的孩子节点的最优拆分点时,
所述根据所述操作任务对所述决策树模型进行操作包括:
按照所述拆分规则生成所述目标节点的新孩子节点;
所述将所述模型信息的计算任务分发给与本机设备连接的各分布式执行机,以供各分布式执行机根据各自的本地数据执行所述计算任务的步骤还包括:
将所述新孩子节点在预设数据特征下各拆分点的数据统计任务发送给各分布式执行机,以供各分布式执行机计算各自...
【专利技术属性】
技术研发人员:李诗琦,黄启军,唐兴兴,林冰垠,王跃,
申请(专利权)人:深圳前海微众银行股份有限公司,
类型:发明
国别省市:广东;44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。