一种基于特征选择的设备风险预测方法及装置

本发明公开了一种基于特征选择的设备风险预测方法及装置，包括：获取设备风险预测任务和设备运行数据，对设备运行数据进行特征提取获取若干个运行特征；基于多层感知机、Gumbel‑Softmax算法和正则化损失构建门控模块，基于门控模块获取各个运行特征的门控参数；基于门控参数对运行特征进行分类，获取第一特征向量和第二特征向量；基于设备风险预测任务的任务类型、第一特征向量和第二特征向量计算预测值，并基于预测值完成设备风险预测任务。本发明通过对设备运行数据进行多维特征提取，并基于所得门控参数对运行特征进行精确分类，最终基于优化后的特征表示进行预测，提高设备风险预测任务的预测准确性。