Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
|
@@ -163,7 +163,7 @@ def plot_mse_heatmap(mse_values, title, df):
|
|
| 163 |
tick_locations = np.linspace(0, len(mse_values) - 1, num_ticks).astype(int)
|
| 164 |
|
| 165 |
# Ensure tick_locations are within bounds
|
| 166 |
-
tick_locations
|
| 167 |
|
| 168 |
tick_labels = [df['Timecode'].iloc[i] if i < len(df) else '' for i in tick_locations]
|
| 169 |
|
|
|
|
| 163 |
tick_locations = np.linspace(0, len(mse_values) - 1, num_ticks).astype(int)
|
| 164 |
|
| 165 |
# Ensure tick_locations are within bounds
|
| 166 |
+
tick_locations = tick_locations[tick_locations < len(df)]
|
| 167 |
|
| 168 |
tick_labels = [df['Timecode'].iloc[i] if i < len(df) else '' for i in tick_locations]
|
| 169 |
|