Direction9/model/__init__.py

"""
model 包 - 模型表示与谱计算

模块:
- spectrum: 谱计算核心（有效秩、Gram 矩阵、Wasserstein 距离）
- extractor: 模型表示提取器（从预存文件加载或从 HuggingFace 模型提取）
"""

from .spectrum import (
    compute_gram_spectrum,
    compute_spectrum_array,
    normalize_representations,
    wasserstein1_distance,
    compute_spectrum_distance_matrix,
    effective_rank_from_eigvals,
    compute_top_k_ratio,
    spectrum_summary,
)

from .extractor import (
    RepresentationLoader,
    RepresentationMeta,
    load_H,
    load_all_H,
    save_H,
)

__all__ = [
    # spectrum
    "compute_gram_spectrum",
    "compute_spectrum_array",
    "normalize_representations",
    "wasserstein1_distance",
    "compute_spectrum_distance_matrix",
    "effective_rank_from_eigvals",
    "compute_top_k_ratio",
    "spectrum_summary",
    # extractor
    "RepresentationLoader",
    "RepresentationMeta",
    "load_H",
    "load_all_H",
    "save_H",
]