如何使用WGAN-GP生成一维滚动轴承振动数据样本。以西储大学（CWRU）数据集为例，提供一个基于训练好的权重参数文件进行测试的代码。

用户12490569

修改于 2026-07-20 10:24:27

170

概述

使用WGAN-GP生成一维滚动轴承振动数据样本。以西储大学（CWRU）数据集为例，提供一个基于训练好的权重参数文件进行测试的代码。WGAN-GP-1D轴承振动数据样本生成方法，西储大学数据集为例，可替换自己的数据。代码注释清楚，包含训练过程的代码train_gan和基于训练好的权重参数文件进行测试的代码generate_gan。可以生成指定的故障类型数据，不平衡比100，可自己调节。环境可见env

文章被收录于专栏：用户12490569的专栏用户12490569的专栏

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

数据集成