Spaces:
Runtime error
Runtime error
Update visualization.py
Browse files- visualization.py +2 -2
visualization.py
CHANGED
|
@@ -180,11 +180,11 @@ def plot_combined_mse(df, mse_embeddings, mse_posture, mse_voice, title, anomaly
|
|
| 180 |
|
| 181 |
# Plot median baseline for each series
|
| 182 |
median = np.median(mse_values)
|
| 183 |
-
ax.axhline(y=median, color=color, linestyle='
|
| 184 |
|
| 185 |
# Plot threshold for each series
|
| 186 |
threshold = np.mean(mse_values) + anomaly_threshold * np.std(mse_values)
|
| 187 |
-
ax.axhline(y=threshold, color=color, linestyle='
|
| 188 |
|
| 189 |
# Plot anomalies in red
|
| 190 |
anomalies = mse_values > threshold
|
|
|
|
| 180 |
|
| 181 |
# Plot median baseline for each series
|
| 182 |
median = np.median(mse_values)
|
| 183 |
+
ax.axhline(y=median, color=color, linestyle=':', alpha=0.5, label=f'{label} Baseline Median')
|
| 184 |
|
| 185 |
# Plot threshold for each series
|
| 186 |
threshold = np.mean(mse_values) + anomaly_threshold * np.std(mse_values)
|
| 187 |
+
ax.axhline(y=threshold, color=color, linestyle='--', alpha=1, label=f'{label} Anomaly Threshold')
|
| 188 |
|
| 189 |
# Plot anomalies in red
|
| 190 |
anomalies = mse_values > threshold
|