Tejasva commited on
Commit Β·
9d46037
1
Parent(s): d24e4fa
fix: remove START prints and emoji prints from train_agent, convert to STEP format
Browse files- inference.py +6 -10
inference.py
CHANGED
|
@@ -323,7 +323,7 @@ def train_agent(num_episodes: int = 10) -> AgentState:
|
|
| 323 |
agent_state = AgentState()
|
| 324 |
all_episodes = []
|
| 325 |
|
| 326 |
-
print(f"
|
| 327 |
logger.info(f"[TRAINING START] Starting RL training for {num_episodes} episodes with model: {MODEL_NAME}")
|
| 328 |
|
| 329 |
# RUN BASELINE FIRST
|
|
@@ -366,7 +366,7 @@ def train_agent(num_episodes: int = 10) -> AgentState:
|
|
| 366 |
break
|
| 367 |
|
| 368 |
# PERFORMANCE COMPARISON
|
| 369 |
-
print("
|
| 370 |
logger.info(f"\n{'='*60}")
|
| 371 |
logger.info(f"[TRAINING SUMMARY]")
|
| 372 |
logger.info(f"{'='*60}")
|
|
@@ -388,17 +388,13 @@ def train_agent(num_episodes: int = 10) -> AgentState:
|
|
| 388 |
|
| 389 |
# Performance classification
|
| 390 |
if improvement > 50:
|
| 391 |
-
print("
|
| 392 |
-
logger.info("π EXCELLENT: Significant improvement over baseline!")
|
| 393 |
elif improvement > 20:
|
| 394 |
-
print("
|
| 395 |
-
logger.info("β
GOOD: Measurable improvement achieved")
|
| 396 |
elif improvement > 0:
|
| 397 |
-
print("
|
| 398 |
-
logger.info("π POSITIVE: Some improvement over baseline")
|
| 399 |
else:
|
| 400 |
-
print("
|
| 401 |
-
logger.info("β οΈ NO IMPROVEMENT: RL agent not better than baseline")
|
| 402 |
|
| 403 |
return agent_state
|
| 404 |
|
|
|
|
| 323 |
agent_state = AgentState()
|
| 324 |
all_episodes = []
|
| 325 |
|
| 326 |
+
print(f"STEP: RL training started - episodes={num_episodes} model={MODEL_NAME}")
|
| 327 |
logger.info(f"[TRAINING START] Starting RL training for {num_episodes} episodes with model: {MODEL_NAME}")
|
| 328 |
|
| 329 |
# RUN BASELINE FIRST
|
|
|
|
| 366 |
break
|
| 367 |
|
| 368 |
# PERFORMANCE COMPARISON
|
| 369 |
+
print("STEP: Training summary analysis")
|
| 370 |
logger.info(f"\n{'='*60}")
|
| 371 |
logger.info(f"[TRAINING SUMMARY]")
|
| 372 |
logger.info(f"{'='*60}")
|
|
|
|
| 388 |
|
| 389 |
# Performance classification
|
| 390 |
if improvement > 50:
|
| 391 |
+
print("STEP: EXCELLENT - significant improvement over baseline")
|
|
|
|
| 392 |
elif improvement > 20:
|
| 393 |
+
print("STEP: GOOD - measurable improvement achieved")
|
|
|
|
| 394 |
elif improvement > 0:
|
| 395 |
+
print("STEP: POSITIVE - some improvement over baseline")
|
|
|
|
| 396 |
else:
|
| 397 |
+
print("STEP: NO IMPROVEMENT - RL agent not better than baseline")
|
|
|
|
| 398 |
|
| 399 |
return agent_state
|
| 400 |
|