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 |
|