Spaces:
Running
Running
Update evaluate_performance.py
Browse files- 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
|
| 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 |
|