fix quantization of seismic
2 changed files+5−3
.gitignoremodified+3−1View file
@@ -6,4 +6,6 @@ __pycache__
66 *.npy
77 *.segy
88 *.mat
9-*.edf
9+*.edf
10+*.zip
11+data
benchcompress/src/benchcompress/datasets/seismic/__init__.pymodified+2−2View file
@@ -60,7 +60,7 @@ def _load_04A_04B_seismic_data() -> np.ndarray:
6060
6161 def _load_quantized_04A_04B_seismic_data():
6262 X = _load_04A_04B_seismic_data()
63- step = 10000
63+ step = 10
6464 X = np.round(X / step).astype(np.int32)
6565 return X
6666
@@ -106,7 +106,7 @@ datasets = [
106106 },
107107 {
108108 "name": "seismic-04A-04B-quantized",
109- "version": "1",
109+ "version": "2",
110110 "description": "Seismic data from Roger Revelle voyage RR1508, quantized.",
111111 "create": lambda: _load_quantized_04A_04B_seismic_data(),
112112 "tags": tags_integer,