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,
|