From 9326505fca1db0437a2630224325d5c0c1ca7e21 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Fri, 13 Jun 2025 14:49:32 -0400 Subject: [PATCH] added changes to datagen --- scripts/dataset_building/data_gen.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/scripts/dataset_building/data_gen.py b/scripts/dataset_building/data_gen.py index 463cb3a..ed2e88a 100644 --- a/scripts/dataset_building/data_gen.py +++ b/scripts/dataset_building/data_gen.py @@ -2,6 +2,7 @@ from utils.data import Recording import numpy as np from utils.signal import block_generator import argparse +import os mods = { "bpsk": {"num_bits_per_symbol": 1, "constellation_type": "psk"}, @@ -61,9 +62,11 @@ def generate_modulated_signals(output_dir): # view if you want # output_recording.view() - + + fname = f"{modulation}_snr{int(snr)}_{i}.npy" + full_path = os.path.join(output_dir, fname) # save to file - output_recording.to_npy() # optionally add path and filename parameters + output_recording.to_npy(path = full_path) # optionally add path and filename parameters if __name__ == "__main__":