diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-02-15 18:19:50 +0000 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-02-15 18:19:50 +0000 |
| commit | c90b48e3f8da9dd0f8d2ae82ddf977436bb0cfc3 (patch) | |
| tree | 43edac8013fec4e65a0b9cddec5314489b4aafc2 /hag/__init__.py | |
Core Hopfield retrieval module with energy-based convergence guarantees,
memory bank, FAISS baseline retriever, evaluation metrics, and end-to-end
pipeline. All 45 tests passing on CPU with synthetic data.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Diffstat (limited to 'hag/__init__.py')
| -rw-r--r-- | hag/__init__.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/hag/__init__.py b/hag/__init__.py new file mode 100644 index 0000000..18496e9 --- /dev/null +++ b/hag/__init__.py @@ -0,0 +1,21 @@ +"""HAG: Hopfield-Augmented Generation.""" + +from hag.config import ( + EncoderConfig, + GeneratorConfig, + HopfieldConfig, + MemoryBankConfig, + PipelineConfig, +) +from hag.datatypes import HopfieldResult, PipelineResult, RetrievalResult + +__all__ = [ + "HopfieldConfig", + "MemoryBankConfig", + "EncoderConfig", + "GeneratorConfig", + "PipelineConfig", + "HopfieldResult", + "RetrievalResult", + "PipelineResult", +] |
