RohitChandramouli6618 commited on
Commit
9eafd01
·
1 Parent(s): c454091

Fix grader weights hospital=0.40/containment=0.35, fix memory threshold in update_memory

Browse files
Files changed (2) hide show
  1. core/policy_update.py +1 -1
  2. server/grader.py +2 -2
core/policy_update.py CHANGED
@@ -51,7 +51,7 @@ def update_memory(
51
 
52
  stored = 0
53
  for step_data in trajectory:
54
- if step_data["reward"] > 0:
55
  memory.store(step_data["obs"], step_data["action"], step_data["reward"])
56
  stored += 1
57
 
 
51
 
52
  stored = 0
53
  for step_data in trajectory:
54
+ if step_data["reward"] > -0.3:
55
  memory.store(step_data["obs"], step_data["action"], step_data["reward"])
56
  stored += 1
57
 
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.45 +
110
- hospital_score * 0.30 +
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.30 +
110
+ hospital_score * 0.45 +
111
  efficiency_score * 0.15 +
112
  speed_score * 0.10
113
  )), 4)