michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

added changes to datagen

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-13 14:49:32 -04:00
parent ec97c78535
commit 9326505fca
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/dataset_building/data_gen.py
View File

@ -2,6 +2,7 @@ from utils.data import Recording
import numpy as np import numpy as np
from utils.signal import block_generator from utils.signal import block_generator
import argparse import argparse
import os
mods = { mods = {
"bpsk": {"num_bits_per_symbol": 1, "constellation_type": "psk"}, "bpsk": {"num_bits_per_symbol": 1, "constellation_type": "psk"},
@ -61,9 +62,11 @@ def generate_modulated_signals(output_dir):
# view if you want # view if you want
# output_recording.view() # output_recording.view()
fname = f"{modulation}_snr{int(snr)}_{i}.npy"
full_path = os.path.join(output_dir, fname)
# save to file # 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__": if __name__ == "__main__":