plot cross_section at center of axis

AI4SIM · Feb 3, 2025 · 7a65153 · 7a65153
1 parent 910c1e0
commit 7a65153
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reactive-flows/cnf-combustion/gnns/models.py b/reactive-flows/cnf-combustion/gnns/models.py
@@ -139,7 +139,7 @@ def on_test_epoch_end(self) -> None:
             self.plotter = plotters.Plotter(
                 self.model.__class__.__name__, plots_path, self.grid_shape
             )
-            self.plotter.cross_section((self.ys.shape[1] - 1), self.ys, self.y_hats)
+            self.plotter.cross_section((self.ys.shape[1] // 2), self.ys, self.y_hats)
             self.plotter.dispersion_plot(self.ys, self.y_hats)
             self.plotter.histo(self.ys, self.y_hats)
             self.plotter.histo2d(self.ys, self.y_hats)