quantized ieeg
1 changed file+26−3
benchcompress/src/benchcompress/datasets/ieeg/__init__.pymodified+26−3View file
@@ -15,7 +15,9 @@ def _load_long_description():
1515
1616 LONG_DESCRIPTION = _load_long_description()
1717
18-tags = ["real", "ecephys", "ieeg", "timeseries", "1d", "float", "continuous"]
18+tags = ["real", "ecephys", "ieeg", "timeseries", "1d", "continuous"]
19+tags_float = tags + ["float"]
20+tags_integer = tags + ["integer"]
1921
2022
2123 def _load_ieeg_openneuro_005592() -> np.ndarray:
@@ -49,14 +51,35 @@ def _load_ieeg_openneuro_005592() -> np.ndarray:
4951 return concatenated
5052
5153
54+def _load_quantized_ieeg_openneuro_005592() -> np.ndarray:
55+ """Load and quantize data from OpenNeuro dataset 005592.
56+
57+ Returns:
58+ Array containing quantized data from first 10 channels concatenated
59+ """
60+ X = _load_ieeg_openneuro_005592()
61+ step = 1
62+ X = np.round(X / step).astype(np.int32)
63+ return X
64+
65+
5266 datasets = [
5367 {
5468 "name": "ieeg-005592",
5569 "version": "1",
5670 "description": "iEEG recording from OpenNeuro dataset ds005592, first 10 channels concatenated.",
5771 "create": lambda: _load_ieeg_openneuro_005592(),
58- "tags": tags,
72+ "tags": tags_float,
73+ "source_file": SOURCE_FILE,
74+ "long_description": LONG_DESCRIPTION,
75+ },
76+ {
77+ "name": "ieeg-005592-quantized",
78+ "version": "1",
79+ "description": "iEEG recording from OpenNeuro dataset ds005592, first 10 channels concatenated, quantized.",
80+ "create": lambda: _load_quantized_ieeg_openneuro_005592(),
81+ "tags": tags_integer,
5982 "source_file": SOURCE_FILE,
6083 "long_description": LONG_DESCRIPTION,
61- }
84+ },
6285 ]