wi-lab commited on
Commit
9aa4468
·
verified ·
1 Parent(s): ad67e90

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -294,7 +294,7 @@ By selecting either `cls_emb` or `channel_emb`, you leverage the pre-trained mod
294
  If your dataset requires labels, you can easily generate them using DeepMIMO data. Here's an example to create labels for either LoS/NLoS classification or beam prediction, depending on the scenario selected:
295
  ```python
296
  from input_preprocess import create_labels
297
- tasks = ['LoS/NLoS Classifcation', 'Beam Prediction']
298
  task = tasks[1] # Choose 0 for LoS/NLoS labels or 1 for beam prediction labels.
299
  labels = create_labels(task, selected_scenario_names, n_beams=64) # For beam prediction, n_beams specifies the number of beams in the codebook. If you're generating labels for LoS/NLoS classification, you can leave this value unchanged as it doesn't impact the label generation.
300
  ```
 
294
  If your dataset requires labels, you can easily generate them using DeepMIMO data. Here's an example to create labels for either LoS/NLoS classification or beam prediction, depending on the scenario selected:
295
  ```python
296
  from input_preprocess import create_labels
297
+ tasks = ['LoS/NLoS Classification', 'Beam Prediction']
298
  task = tasks[1] # Choose 0 for LoS/NLoS labels or 1 for beam prediction labels.
299
  labels = create_labels(task, selected_scenario_names, n_beams=64) # For beam prediction, n_beams specifies the number of beams in the codebook. If you're generating labels for LoS/NLoS classification, you can leave this value unchanged as it doesn't impact the label generation.
300
  ```