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 /configs | |
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 'configs')
| -rw-r--r-- | configs/2wikimhqa.yaml | 22 | ||||
| -rw-r--r-- | configs/default.yaml | 21 | ||||
| -rw-r--r-- | configs/hotpotqa.yaml | 22 | ||||
| -rw-r--r-- | configs/musique.yaml | 22 |
4 files changed, 87 insertions, 0 deletions
diff --git a/configs/2wikimhqa.yaml b/configs/2wikimhqa.yaml new file mode 100644 index 0000000..0048cf2 --- /dev/null +++ b/configs/2wikimhqa.yaml @@ -0,0 +1,22 @@ +hopfield: + beta: 2.0 + max_iter: 8 + conv_threshold: 1.0e-4 + top_k: 5 + +memory: + embedding_dim: 768 + normalize: true + +encoder: + model_name: "facebook/contriever-msmarco" + max_length: 512 + batch_size: 64 + +generator: + model_name: "meta-llama/Llama-3.1-8B-Instruct" + max_new_tokens: 128 + temperature: 0.0 + +retriever_type: "hopfield" +dataset: "2wikimultihopqa" diff --git a/configs/default.yaml b/configs/default.yaml new file mode 100644 index 0000000..cb76d34 --- /dev/null +++ b/configs/default.yaml @@ -0,0 +1,21 @@ +hopfield: + beta: 1.0 + max_iter: 5 + conv_threshold: 1.0e-4 + top_k: 5 + +memory: + embedding_dim: 768 + normalize: true + +encoder: + model_name: "facebook/contriever-msmarco" + max_length: 512 + batch_size: 64 + +generator: + model_name: "meta-llama/Llama-3.1-8B-Instruct" + max_new_tokens: 128 + temperature: 0.0 + +retriever_type: "hopfield" diff --git a/configs/hotpotqa.yaml b/configs/hotpotqa.yaml new file mode 100644 index 0000000..99ea9d7 --- /dev/null +++ b/configs/hotpotqa.yaml @@ -0,0 +1,22 @@ +hopfield: + beta: 2.0 + max_iter: 8 + conv_threshold: 1.0e-4 + top_k: 5 + +memory: + embedding_dim: 768 + normalize: true + +encoder: + model_name: "facebook/contriever-msmarco" + max_length: 512 + batch_size: 64 + +generator: + model_name: "meta-llama/Llama-3.1-8B-Instruct" + max_new_tokens: 128 + temperature: 0.0 + +retriever_type: "hopfield" +dataset: "hotpotqa" diff --git a/configs/musique.yaml b/configs/musique.yaml new file mode 100644 index 0000000..52dd7c4 --- /dev/null +++ b/configs/musique.yaml @@ -0,0 +1,22 @@ +hopfield: + beta: 2.0 + max_iter: 8 + conv_threshold: 1.0e-4 + top_k: 5 + +memory: + embedding_dim: 768 + normalize: true + +encoder: + model_name: "facebook/contriever-msmarco" + max_length: 512 + batch_size: 64 + +generator: + model_name: "meta-llama/Llama-3.1-8B-Instruct" + max_new_tokens: 128 + temperature: 0.0 + +retriever_type: "hopfield" +dataset: "musique" |