ModuFlow/mask-ddpm
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

连续型特征在时许相关性上的不足

Browse Source
This commit is contained in:
MingzheYang
2026-01-23 15:06:52 +08:00
parent 0d17be9a1c
commit ff12324560
12 changed files with 1212 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/run_pipeline.py
View File

@@ -48,6 +48,8 @@ def main():
seq_len = cfg.get("sample_seq_len", cfg.get("seq_len", 64))
batch_size = cfg.get("sample_batch_size", cfg.get("batch_size", 2))
clip_k = cfg.get("clip_k", 5.0)
data_glob = cfg.get("data_glob", "")
data_path = cfg.get("data_path", "")
run([sys.executable, str(base_dir / "prepare_data.py")])
run([sys.executable, str(base_dir / "train.py"), "--config", args.config, "--device", args.device])
run(
@@ -70,7 +72,11 @@ def main():
"--use-ema",
]
)
run([sys.executable, str(base_dir / "evaluate_generated.py")])
ref = data_glob if data_glob else data_path
if ref:
run([sys.executable, str(base_dir / "evaluate_generated.py"), "--reference", str(ref)])
else:
run([sys.executable, str(base_dir / "evaluate_generated.py")])
run([sys.executable, str(base_dir / "plot_loss.py")])