Fix Bug

app.py

@@ -1286,7 +1286,7 @@ def run_model(model_name):
                         )
                     # Transponer la matriz para alinear correctamente los ejes
                     # Transponer y limpiar valores inválidos (NaN, inf, -inf)
-                    heatmap_data = heat_stat.T
+                    # heatmap_data = heat_stat.T
                     heatmap_data = np.nan_to_num(heat_stat.T, nan=0.0, posinf=0.0, neginf=0.0)
                     
                     # Definir el color mapper
@@ -1509,6 +1509,7 @@ def run_model(model_name):
 
                 # Rango real de los datos
                 df_heatmap = df_heatmap_base.copy()
+                # Rango real de los datos
                 x_min, x_max = df_heatmap[x_comp].min(), df_heatmap[x_comp].max()
                 y_min, y_max = df_heatmap[y_comp].min(), df_heatmap[y_comp].max()
 
@@ -1521,8 +1522,9 @@ def run_model(model_name):
                 grid_size = st.slider("Grid size (resolución del heatmap)", min_value=10, max_value=100, value=40, step=5)
                 x_bins = np.linspace(x_min, x_max, grid_size + 1)
                 y_bins = np.linspace(y_min, y_max, grid_size + 1)
-                x_range = (x_min - padding, x_max + padding)
-                y_range = (y_min - padding, y_max + padding)
+
+                # Usar los mismos valores para figure y image
+                x_range, y_range = (x_min, x_max), (y_min, y_max)
                 
                 
                 df_heatmap["x"] = df_heatmap[x_comp]
@@ -1542,7 +1544,7 @@ def run_model(model_name):
                     df_heatmap['x'], df_heatmap['y'], df_heatmap[selected_feature],
                     statistic='mean', bins=[x_bins, y_bins]
                 )
-                heatmap_data = heat_stat.T
+                # heatmap_data = heat_stat.T
                 heatmap_data = np.nan_to_num(heat_stat.T, nan=0.0, posinf=0.0, neginf=0.0)
                 
                 cmap = plt.get_cmap("RdYlGn")