Log issue

unda-ml · Feb 1, 2024 · 4cfdf96 · 4cfdf96
1 parent 47f32f3
commit 4cfdf96
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/core/network.rs b/src/core/network.rs
@@ -311,9 +311,11 @@ impl Network{
         }
 
         self.loss = self.loss_train[self.loss_train.len() - 1];
-        println!("Trained to a loss of {:.2}%", self.loss * 100.0);
-        for i in 0..self.layers.len() - 1 {
-            println!("Error on layer {}: +/- {:.2}", i + 1, self.layers[i].get_loss());
+        if self.log {
+            println!("Trained to a loss of {:.2}%", self.loss * 100.0);
+            for i in 0..self.layers.len() - 1 {
+                println!("Error on layer {}: +/- {:.2}", i + 1, self.layers[i].get_loss());
+            }
         }
     }