Spaces:
Sleeping
Sleeping
File size: 1,065 Bytes
fc58aef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | from dataclasses import dataclass, field
from typing import Optional, Dict, Any
@dataclass
class EchoAction:
"""Action: model's response with embedded confidence and answer."""
response: str # Full response text containing <confidence> and <answer> tags
@dataclass
class EchoObservation:
"""Observation returned after each step."""
question: str
domain: str
difficulty: str
reward: float
accuracy: float
confidence: int
brier_score: float
ece: float
is_correct: bool
thinking: str = ""
feedback: str = ""
episode_step: int = 0
total_steps: int = 0
@dataclass
class EchoState:
"""Full environment state."""
current_question: str = ""
domain: str = ""
difficulty: str = ""
phase: int = 1
step_count: int = 0
total_reward: float = 0.0
accuracy_history: list = field(default_factory=list)
confidence_history: list = field(default_factory=list)
ece_history: list = field(default_factory=list)
domain_stats: Dict[str, Any] = field(default_factory=dict)
|