ott.solvers.linear.semidiscrete.SemidiscreteOutput.marginal_chi2_error

ott.solvers.linear.semidiscrete.SemidiscreteOutput.marginal_chi2_error#

SemidiscreteOutput.marginal_chi2_error(rng, *, num_iters, batch_size)[source]#

Compute the marginal chi-squared error.

Parameters:
  • rng (Array) – Random key used for seeding.

  • num_iters (int) – Number of iterations used to estimate the error.

  • batch_size (int) – Number of points to sample from the source distribution at each iteration.

Return type:

Array

Returns:

The marginal chi-squared error.