Update input_preprocess.py

input_preprocess.py

@@ -359,4 +359,11 @@ def label_prepend(deepmimo_data, preprocessed_chs, task, scenario_idxs, n_beams=
     
     preprocessed_chs = [preprocessed_chs[i] + [labels[i]] for i in range(len(preprocessed_chs))]
     
-    return preprocessed_chs
+    return preprocessed_chs
+
+def create_labels(task, scenario_names, n_beams=64):
+    labels = []
+    for scenario_name in scenario_names:
+        data = DeepMIMO_data_gen(scenario_name)
+        labels.extend(label_gen(task, data, scenario_name, n_beams=n_beams))
+    return labels