[发明专利]模型训练方法、感染人数预测方法、介质及设备

说明书

本公开提供一种模型训练方法、感染人数预测方法、存储介质及电子设备；涉及人工智能技术领域。所述方法包括：获取目标传染病的第一特征数据，所述第一特征数据包括所述目标传染病的多个影响因素集合和感染人数；根据各所述影响因素集合和所述感染人数构建与各所述影响因素集合对应的感染人数预测模型；利用所述多个影响因素集合和所述感染人数对各所述感染人数预测模型进行训练，得到非标注数据上方差最小的各所述感染人数预测模型，并利用训练好的各所述感染人数预测模型预测该传染病在目标时间段内的感染人数。本公开可以提升感染人数预测模型的预测性能，进而提高感染人数预测的准确性。

技术领域

本公开涉及人工智能技术领域，具体而言，涉及一种模型训练方法、感染人数预测方法、计算机可读存储介质以及电子设备。

背景技术

当某一地区发生传染病时，准确地预测传染病的感染人数对于控制传染病具有重要意义。比如通过预测感染人数，便于提前做好疾病预防和医疗救助措施，从而减缓或阻止疾病的传播。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

发明内容

本公开提供一种模型训练方法、感染人数预测方法、计算机可读存储介质以及电子设备，进而至少在一定程度上克服由于感染人数预测模型的预测性能较差而导致感染人数预测的准确性较低的问题。

根据本公开的第一方面，提供一种模型训练方法，包括：

获取目标传染病的第一特征数据，所述第一特征数据包括所述目标传染病的多个影响因素集合和感染人数；

根据各所述影响因素集合和所述感染人数构建与各所述影响因素集合对应的感染人数预测模型；

利用所述多个影响因素集合和所述感染人数对各所述感染人数预测模型进行训练，得到非标注数据上方差最小的各所述感染人数预测模型。

在本公开的一种示例性实施例中，获取目标传染病的第一特征数据之前，所述方法还包括：

获取所述目标传染病的原始特征数据，所述原始特征数据包括所述目标传染病的多个影响因素和所述感染人数；

基于所述多个影响因素和所述感染人数通过启发式搜索方式确定所述多个影响因素集合。

在本公开的一种示例性实施例中，所述多个影响因素集合至少包括第一影响因素集合和第二影响因素集合；

所述基于所述多个影响因素和所述感染人数通过启发式搜索方式确定所述多个影响因素集合，包括：

构造第一空集和第二空集；

轮询各所述影响因素，计算各所述影响因素、所述感染人数、所述第一空集和所述第二空集之间的关联性；

确定所述关联性满足预设关联条件的目标影响因素，并根据所述目标影响因素更新所述第一空集和所述第二空集，得到所述第一影响因素集合和所述第二影响因素集合。

在本公开的一种示例性实施例中，所述根据各所述影响因素集合和所述感染人数构建与各所述影响因素集合对应的感染人数预测模型，包括：

根据各所述影响因素集合和所述感染人数之间的对应关系得到各所述影响因素集合对应的标注数据和非标注数据；

根据各所述影响因素集合对应的所述标注数据确定所述标注数据服从的第一高斯分布；

根据各所述影响因素集合对应的所述非标注数据确定所述非标注数据服从的第二高斯分布；

对所述第一高斯分布和所述第二高斯分布进行联合分布计算，得到各所述感染人数预测模型。