一种用于轨道交通故障诊断的类别识别模型训练方法、故障类别识别方法

本公开提供了一种用于轨道交通故障诊断的类别识别模型训练方法、故障类别识别方法，可以应用于轨道交通故障诊断技术领域。该方法包括：获取训练样本，其中，训练样本包括样本故障特征数据、样本故障类别标签，样本故障特征数据表征在轨道交通场景中采集的故障特征数据；对样本故障特征数据进行过采样处理，得到样本合成特征数据，其中，样本合成特征数据的数量大于预设数量阈值；利用生成对抗网络处理样本合成特征数据和样本故障特征数据，得到样本目标生成数据；利用初始类别识别模型处理样本目标生成数据，得到样本故障类别识别结果；基于样本故障类别识别结果和样本故障类别标签，训练初始类别识别模型，得到训练后的类别识别模型。