added plotting for best model by default

tkchafin · May 9, 2024 · 7404029 · 7404029
1 parent 02b56a5
commit 7404029
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/resistnet/model_optimisation.py b/src/resistnet/model_optimisation.py
@@ -497,6 +497,12 @@ def model_average(self, out="", plot=True):
             if matrix_r is not None:
                 matrix_avg += matrix_r * weight
 
+            if model_num == 0:
+                oname = f"{out}.Model-Best"
+                self.resistance_network.output_and_plot_model(
+                    oname, matrix_r, edge_r
+                ) 
+
             if self.report_all:
                 oname = f"{out}.Model-{model_num}"
                 self.resistance_network.output_and_plot_model(