/ concept-collection / ephys_compression_tests
concept-collection / ephys_compression_tests
update algs
Jeremy Magland <jmagland@flatironinstitute.org> committed commit c8e075b13f72 parent 5be77d1 Browse files
1 changed file+2−2
python/ephys_compression_tests/algorithms/ans/__init__.pymodified+2−2View file
@@ -256,7 +256,7 @@ for a in algorithm_dicts_base:
256256 })
257257
258258 # Add lossy ar2
259-for tolerance in [1, 2, 3]:
259+for tolerance in [1, 2, 3, 4, 5]:
260260 def encode0_ar2_lossy(x: np.ndarray, tolerance=tolerance) -> bytes:
261261 coeffs, residuals, initial_values = encode_ar_lossy(x, order=2, step=tolerance * 2 + 1)
262262 encoded_residuals = ans_encode_0(residuals)
@@ -277,7 +277,7 @@ for tolerance in [1, 2, 3]:
277277 reconstructed = decode_ar(coeffs, residuals, initial_values)
278278 return reconstructed.reshape(shape)
279279 algorithm_dicts.append({
280- "name": f"ans-ar2-lossy-{tolerance}",
280+ "name": f"ans-ar2-lossy-tol{tolerance}",
281281 "version": "1",
282282 "encode": encode0_ar2_lossy,
283283 "decode": decode0_ar2_lossy,