reab5555 commited on
Commit
a337f54
·
verified ·
1 Parent(s): 77142e3

Update visualization.py

Browse files
Files changed (1) hide show
  1. 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='--', alpha=0.5, label=f'{label} 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=0.5, label=f'{label} Threshold')
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