Upload input_preprocess.py
Browse files- input_preprocess.py +3 -1
input_preprocess.py
CHANGED
@@ -171,11 +171,13 @@ def tokenizer(selected_scenario_names,
|
|
171 |
MAX_LEN=513,
|
172 |
masking_percent=.40,
|
173 |
mask=False,
|
|
|
174 |
seed=42,
|
175 |
snr=None):
|
176 |
|
177 |
patches, labels, raw_chs = patch_gen(
|
178 |
-
selected_scenario_names=selected_scenario_names,
|
|
|
179 |
bs_idxs=bs_idxs,
|
180 |
load_data=load_data,
|
181 |
task=task,
|
|
|
171 |
MAX_LEN=513,
|
172 |
masking_percent=.40,
|
173 |
mask=False,
|
174 |
+
manual_data=None,
|
175 |
seed=42,
|
176 |
snr=None):
|
177 |
|
178 |
patches, labels, raw_chs = patch_gen(
|
179 |
+
selected_scenario_names=selected_scenario_names,
|
180 |
+
manual_data=None,
|
181 |
bs_idxs=bs_idxs,
|
182 |
load_data=load_data,
|
183 |
task=task,
|