diff options
| -rw-r--r-- | src/experiment.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/experiment.py b/src/experiment.py index dd2033c..b508438 100644 --- a/src/experiment.py +++ b/src/experiment.py @@ -54,10 +54,10 @@ class ExperimentMetrics: self.test_accuracies = [] def add_train_loss(self, loss): - self.train_losses.append(loss) + self.train_losses.append(round(loss.tolist(), 3)) def add_test_metrics(self, loss, accuracy): - self.test_losses.append(loss) + self.test_losses.append(round(loss.tolist(), 3)) self.test_accuracies.append(accuracy) def get_dataframe(self): |