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
|