一种镁合金力学性能预测方法

本发明涉及一种镁合金力学性能预测方法，包括如下步骤：步骤1：获取原始数据集，构建初始网络，初始网络为带有自适应激活函数机制的KAN网络，利用原始数据集对初始网络进行预训练，得到初始预测模型；步骤2：构建数据增强模型，数据增强模型为WGAN网络，将原始数据输入数据增强模型获取扩展数据；步骤3：将原始数据与扩展数据合并，形成增强数据集；步骤4：利用增强数据集训练初始预测模型，得到性能预测模型；步骤5：利用性能预测模型预测镁合金的力学性能。本发明结合了自适应激活函数机制和WGAN网络的优势，不仅能够有效捕捉镁合金力学性能中的复杂非线性关系，还通过数据增强缓解了小样本数据下的过拟合问题。