jax_privacy.matrix_factorization.buffered_toeplitz.get_init_blt
- jax_privacy.matrix_factorization.buffered_toeplitz.get_init_blt(num_buffers=3, init_blt=None)[source]
Returns an initial BufferedToeplitz for initializing optimization.
Currently, this defaults to BufferedToeplitz.from_rational_approx_to_sqrt_x, setting max_buf_decay and max_pillutla_score so that the solution is within the feasible set imposed by the optimization penalities in LosssFn.compute_penalties. However, this initialization choice may be changed in the future.
- Parameters:
num_buffers (
int) – The number of buffers to use in the initial BLT, greater than or equal to zero.init_blt (
BufferedToeplitz|None) – An initial BLT to use. If None, a default initialization is used. This is a convienence for callers who want to handle an optional explicit initialization and check that it has the correct number of buffers.
- Return type:
- Returns:
A BufferedToeplitz with the requested number of buffers.