From ccff87c15263d1d63235643d54322b991366952e Mon Sep 17 00:00:00 2001 From: LuyaoZhuang Date: Sun, 26 Oct 2025 05:09:57 -0400 Subject: commit --- src/config.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/config.py (limited to 'src/config.py') diff --git a/src/config.py b/src/config.py new file mode 100644 index 0000000..6258a65 --- /dev/null +++ b/src/config.py @@ -0,0 +1,21 @@ +from dataclasses import dataclass +from src.utils import LLM_Model +@dataclass +class LinearRAGConfig: + dataset_name: str + embedding_model: str = "all-mpnet-base-v2" + llm_model: LLM_Model = None + chunk_token_size: int = 1000 + chunk_overlap_token_size: int = 100 + spacy_model: str = "en_core_web_trf" + working_dir: str = "./import" + batch_size: int = 128 + max_workers: int = 16 + retrieval_top_k: int = 5 + max_iterations: int = 3 + top_k_sentence: int = 2 + passage_ratio: float = 1.5 + passage_node_weight: float = 0.05 + damping: float = 0.5 + initial_threshold: float = 0.5 + iteration_threshold: float = 0.5 \ No newline at end of file -- cgit v1.2.3