Metric Ops

Module: generative_models.core.evaluation.metrics.metric_ops

Source: generative_models/core/evaluation/metrics/metric_ops.py

Overview

JAX-compatible operations for metrics computation.

This module provides JAX implementations of common algorithms needed for metrics, replacing numpy and sklearn dependencies to ensure compatibility with NNX modules.

Functions

bincount

def bincount()

compute_cdf

def compute_cdf()

compute_ks_distance

def compute_ks_distance()

corrcoef

def corrcoef()

matrix_sqrtm

def matrix_sqrtm()

nearest_neighbors

def nearest_neighbors()

pairwise_distances

def pairwise_distances()

Module Statistics

• Classes: 0
• Functions: 7
• Imports: 2