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
Files changed (1) hide show
  1. 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"START: RL training for {num_episodes} episodes with 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,7 +366,7 @@ def train_agent(num_episodes: int = 10) -> AgentState:
366
  break
367
 
368
  # PERFORMANCE COMPARISON
369
- print("START: Training summary analysis")
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("πŸš€ EXCELLENT: Significant improvement over baseline!")
392
- logger.info("πŸš€ EXCELLENT: Significant improvement over baseline!")
393
  elif improvement > 20:
394
- print("βœ… GOOD: Measurable improvement achieved")
395
- logger.info("βœ… GOOD: Measurable improvement achieved")
396
  elif improvement > 0:
397
- print("πŸ“ˆ POSITIVE: Some improvement over baseline")
398
- logger.info("πŸ“ˆ POSITIVE: Some improvement over baseline")
399
  else:
400
- print("⚠️ NO IMPROVEMENT: RL agent not better than baseline")
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