Y-debug-sys / Diffusion-TS

[ICLR 2024] Official Implementation of "Diffusion-TS: Interpretable Diffusion for General Time Series Generation"
MIT License
184 stars 26 forks source link

数据集结构 #43

Closed HIUZS closed 4 months ago

HIUZS commented 4 months ago

作者,你好。我想改变训练集ground_truth_24_train.npy与无条件生成fake_24_1.npy文件的样本数量,即.npy数据结构中(n,windows,feature)中n的值,该修改config文件中的哪一个参数或代码中的哪条语句

Y-debug-sys commented 4 months ago

你好,见

https://github.com/Y-debug-sys/Diffusion-TS/blob/21ca525ed6a0ce4322e05d665c792d2b8dc41f99/main.py#L90

num 为目标采样数,size_every 为一轮采样数,谢谢!