qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

added changes to datagen
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 13m1s
Details

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

5
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"},
@ -62,8 +63,10 @@ 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__":