ankanghosh commited on
Commit
4dbfec8
·
verified ·
1 Parent(s): e6d6436

Update evaluate_performance.py

Browse files
Files changed (1) hide show
  1. evaluate_performance.py +1 -1
evaluate_performance.py CHANGED
@@ -424,7 +424,7 @@ def create_charts(metrics, output_dir="."):
424
  plt.savefig(os.path.join(output_dir, 'accuracy_by_class.png'))
425
  plt.close() # Close the figure to free memory
426
 
427
- # Plot 2: Safety Rate by class
428
  plt.figure(figsize=(10, 6))
429
  safety_rates = [metrics['per_class_metrics'][cls]['safety_rate'] for cls in classes]
430
 
 
424
  plt.savefig(os.path.join(output_dir, 'accuracy_by_class.png'))
425
  plt.close() # Close the figure to free memory
426
 
427
+ # Plot 2: Safety rate by class
428
  plt.figure(figsize=(10, 6))
429
  safety_rates = [metrics['per_class_metrics'][cls]['safety_rate'] for cls in classes]
430