RohitChandramouli6618 commited on
Commit
1169cdd
·
1 Parent(s): 4d57df8

Fix: D0 seed above threshold, faster hospital drain, containment as primary metric

Browse files
server/environment.py CHANGED
@@ -260,7 +260,7 @@ class EpidemicContainmentEnv(Environment):
260
  if district.true_infection_rate > INFECTION_THRESHOLD:
261
  # Capacity drains proportional to how far above threshold
262
  excess = district.true_infection_rate - INFECTION_THRESHOLD
263
- drain = round(excess * 0.15, 4)
264
  district.hospital_capacity_remaining = max(
265
  0.0,
266
  district.hospital_capacity_remaining - drain
 
260
  if district.true_infection_rate > INFECTION_THRESHOLD:
261
  # Capacity drains proportional to how far above threshold
262
  excess = district.true_infection_rate - INFECTION_THRESHOLD
263
+ drain = round(excess * 0.25, 4)
264
  district.hospital_capacity_remaining = max(
265
  0.0,
266
  district.hospital_capacity_remaining - drain
server/grader.py CHANGED
@@ -106,8 +106,8 @@ def grade_trajectory(trajectory: List[TrajectoryStep], task_name: str) -> GradeR
106
  # Efficiency (15%) — quality of resource allocation decisions.
107
  # Speed (10%) — tiebreaker rewarding proactive early containment.
108
  final_score = round(min(1.0, max(0.0,
109
- containment_score * 0.30 +
110
- hospital_score * 0.45 +
111
  efficiency_score * 0.15 +
112
  speed_score * 0.10
113
  )), 4)
 
106
  # Efficiency (15%) — quality of resource allocation decisions.
107
  # Speed (10%) — tiebreaker rewarding proactive early containment.
108
  final_score = round(min(1.0, max(0.0,
109
+ containment_score * 0.45 +
110
+ hospital_score * 0.30 +
111
  efficiency_score * 0.15 +
112
  speed_score * 0.10
113
  )), 4)
server/tasks/task_easy.py CHANGED
@@ -22,7 +22,7 @@ class EasyTask(BaseTask):
22
  # With TREATMENT_REDUCTION=0.05 and good strategy, agent contains both
23
  # districts in 6-8 steps, earning a speed bonus. Requires sustained focus
24
  # on D0 first before D1 grows above safe threshold.
25
- seed_infections = [0.40, 0.04]
26
 
27
  return CityState(
28
  day = 0,
 
22
  # With TREATMENT_REDUCTION=0.05 and good strategy, agent contains both
23
  # districts in 6-8 steps, earning a speed bonus. Requires sustained focus
24
  # on D0 first before D1 grows above safe threshold.
25
+ seed_infections = [0.50, 0.06]
26
 
27
  return CityState(
28
  day = 0,