updated workflow scripts

.riahub/workflows/workflow.yaml

@@ -40,7 +40,7 @@ jobs:
       - name: 1. Build HDF5 Dataset
         run: |
          mkdir -p data
-         PYTHONPATH=. python scripts/produce_dataset.py
+         PYTHONPATH=. python data/scripts/produce_dataset.py
          echo "datasets produced successfully"
         shell: bash
 

conf/app.yaml

@@ -3,24 +3,25 @@ general:
   run_mode: prod
 
 dataset:
-  input_dir: "recordings"
+  input_dir: data/recordings
   num_slices: 8
   train_split: 0.8
   val_split : 0.2
-  seed: 42
-  output_dir: "data"
+  seed: 25
+  modulation_types: [bpsk, qpsk, qam16, qam64]
+  output_dir: data/dataset
 
 training:
   batch_size: 64
   epochs: 50
   learning_rate: 0.001
-  checkpoint_path: "checkpoints/model.ckpt"
+  checkpoint_path: checkpoints/model.ckpt
   use_gpu: true
 
 inference:
-  model_path: "checkpoints/model.ckpt"
+  model_path: checkpoints/model.ckpt
   num_classes: 10
-  output_path: "results/output.json"
+  output_path: results/output.json
 
 app:
-  build_dir: "dist"
+  build_dir: dist

helpers/app_settings.py

@@ -14,6 +14,7 @@ class DataSetConfig:
     num_slices: int
     train_split: float
     seed: int
+    modulation_types: list
     val_split: float
     output_dir: str