setu.total_variation#

setu.total_variation(samples_a, samples_b)[source]#

Total variation distance between two discrete sample distributions.

Estimates TV(P, Q) = 0.5 * sum_x abs(P(x) - Q(x)) from empirical PMFs. Works for univariate and multivariate discrete data.

Parameters:

  • samples_a (Array) – First set of integer samples (n, dim) or (n,).

  • samples_b (Array) – Second set of integer samples (m, dim) or (m,).

Return type:

float

Returns:

Total variation distance in [0, 1]. 0 = identical, 1 = disjoint support.