v0.3: config with auto-tune and profile path
Browse files- aria_llm/config.py +15 -9
aria_llm/config.py
CHANGED
|
@@ -1,12 +1,10 @@
|
|
| 1 |
"""
|
| 2 |
-
ARIA Configuration v0.
|
| 3 |
========================
|
| 4 |
|
| 5 |
-
All thresholds and hyperparameters. v0.
|
| 6 |
-
-
|
| 7 |
-
-
|
| 8 |
-
- max_corrections_per_step: budget (default 1 — only strongest signal corrects)
|
| 9 |
-
- correction_scale: global multiplier on all correction strengths (default 0.1)
|
| 10 |
"""
|
| 11 |
|
| 12 |
from dataclasses import dataclass, field
|
|
@@ -17,14 +15,22 @@ from typing import Optional, List, Dict
|
|
| 17 |
class ARIAConfig:
|
| 18 |
"""Configuration for the ARIA reliability attachment.
|
| 19 |
|
| 20 |
-
v0.
|
| 21 |
"""
|
| 22 |
|
| 23 |
-
# === CALIBRATION
|
| 24 |
calibration_steps: int = 20
|
| 25 |
sensitivity_k: float = 2.5
|
| 26 |
|
| 27 |
-
# ===
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
max_corrections_per_step: int = 1
|
| 29 |
correction_scale: float = 0.1
|
| 30 |
|
|
|
|
| 1 |
"""
|
| 2 |
+
ARIA Configuration v0.3
|
| 3 |
========================
|
| 4 |
|
| 5 |
+
All thresholds and hyperparameters. v0.3 adds:
|
| 6 |
+
- auto_tune_correction_scale: automatically set correction_scale from calibration variance
|
| 7 |
+
- calibration_profile_path: load a pre-saved calibration profile (skip calibration phase)
|
|
|
|
|
|
|
| 8 |
"""
|
| 9 |
|
| 10 |
from dataclasses import dataclass, field
|
|
|
|
| 15 |
class ARIAConfig:
|
| 16 |
"""Configuration for the ARIA reliability attachment.
|
| 17 |
|
| 18 |
+
v0.3: Auto-tuning + calibration profiles.
|
| 19 |
"""
|
| 20 |
|
| 21 |
+
# === CALIBRATION ===
|
| 22 |
calibration_steps: int = 20
|
| 23 |
sensitivity_k: float = 2.5
|
| 24 |
|
| 25 |
+
# === AUTO-TUNE (NEW in v0.3) ===
|
| 26 |
+
auto_tune_correction_scale: bool = True # auto-set correction_scale after calibration
|
| 27 |
+
auto_tune_min_scale: float = 0.02 # floor for auto-tuned scale
|
| 28 |
+
auto_tune_max_scale: float = 0.3 # ceiling for auto-tuned scale
|
| 29 |
+
|
| 30 |
+
# === CALIBRATION PROFILES (NEW in v0.3) ===
|
| 31 |
+
calibration_profile_path: Optional[str] = None # load pre-saved profile
|
| 32 |
+
|
| 33 |
+
# === CORRECTION BUDGET ===
|
| 34 |
max_corrections_per_step: int = 1
|
| 35 |
correction_scale: float = 0.1
|
| 36 |
|