Spaces:
Paused
Paused
File size: 2,470 Bytes
a52f96d |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
"""
Shared interfaces for all components.
DO NOT MODIFY - must match teacher and task generator teams.
"""
from dataclasses import dataclass
from typing import List, Dict
from abc import ABC, abstractmethod
@dataclass
class Task:
"""A reading comprehension task."""
passage: str
question: str
choices: List[str] # 4 choices: ['A) ...', 'B) ...', 'C) ...', 'D) ...']
answer: int # Index of correct answer (0-3)
topic: str # e.g., 'history', 'science', 'literature', 'geography', 'current_events'
difficulty: str # 'easy', 'medium', 'hard'
task_id: str
@dataclass
class StudentState:
"""Student's current learning state."""
topic_accuracies: Dict[str, float] # topic -> accuracy (0.0-1.0)
topic_attempts: Dict[str, int] # topic -> number of attempts
time_since_practice: Dict[str, float] # topic -> time since last practice
total_timesteps: int
current_time: float
@dataclass
class TeacherAction:
"""Teacher's decision about what to teach next."""
topic: str
difficulty: str
is_review: bool
class TaskGeneratorInterface(ABC):
@abstractmethod
def generate_task(self, topic: str, difficulty: str) -> Task:
pass
@abstractmethod
def get_available_topics(self) -> List[str]:
pass
@abstractmethod
def get_available_difficulties(self) -> List[str]:
pass
class StudentAgentInterface(ABC):
@abstractmethod
def answer(self, task: Task) -> int:
"""Predict answer to a task (before learning)."""
pass
@abstractmethod
def learn(self, task: Task) -> bool:
"""Learn from a task. Returns True if answer was correct."""
pass
@abstractmethod
def evaluate(self, eval_tasks: List[Task]) -> float:
"""Evaluate on held-out test set. Returns accuracy (0.0-1.0)."""
pass
@abstractmethod
def get_state(self) -> StudentState:
"""Get current state for teacher to observe."""
pass
@abstractmethod
def advance_time(self, delta: float = 1.0):
"""Advance time for forgetting simulation."""
pass
class TeacherAgentInterface(ABC):
@abstractmethod
def select_action(self, student_state: StudentState) -> TeacherAction:
pass
@abstractmethod
def update(self, action: TeacherAction, reward: float):
pass
@abstractmethod
def get_statistics(self) -> Dict:
pass
|