【免费下载】 TimeGAN: 时间序列生成对抗网络代码库

描述

本代码库提供了“时间序列生成对抗网络（TimeGAN）”的实现，该网络由尹成ung、丹尼尔·贾瑞特和米哈埃拉·范德沙尔共同开发。该研究成果发表于2019年的神经信息处理系统（NeurIPS）会议上。

本目录包含了使用一个合成数据集和两个真实数据集生成合成时间序列数据的TimeGAN框架的实现。具体数据集包括：

• 正弦数据：合成数据集，用于验证模型的基本功能。
• 股票数据：真实数据集，用于测试模型在实际应用中的表现。
• 能源数据：真实数据集，用于进一步验证模型的泛化能力。

使用方法

要运行TimeGAN框架的训练和评估管道，只需执行以下命令：

python3 -m main_timegan.py

或者，您也可以在tutorial_timegan.ipynb中查看TimeGAN的Jupyter Notebook教程，以更直观的方式了解模型的运行过程。

注意事项

• 任何模型架构都可以用作生成器和鉴别器模型，例如RNN或Transformers。
• 代码中提供了详细的注释，帮助您理解模型的各个部分。

通过本代码库，您可以深入了解TimeGAN的实现细节，并将其应用于您自己的时间序列数据生成任务中。