From 38896d503cb87bdc340101003c94901d0e4d4a8e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=BChle=2C=20Laura=20Christine=20=28lakue103=29?=
<laura.kuehle@uni-duesseldorf.de>
Date: Tue, 1 Mar 2022 12:59:54 +0100
Subject: [PATCH] Reduced number of testing epochs to 50.
---
ANN_Training.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/ANN_Training.py b/ANN_Training.py
index 65dc77a..158c6f5 100644
--- a/ANN_Training.py
+++ b/ANN_Training.py
@@ -10,7 +10,7 @@ TODO: Fix random seed
TODO: Write-protect all data and models
TODO: Put legend outside plot (bbox_to_anchor)
TODO: Put plotting into separate function
-TODO: Reduce number of testing epochs to 50
+TODO: Reduce number of testing epochs to 50 -> Done
"""
import numpy as np
@@ -205,7 +205,7 @@ class ModelTrainer(object):
Dictionary containing classification evaluation data.
"""
- self.epoch_training(training_set, num_epochs=100, verbose=False)
+ self.epoch_training(training_set, num_epochs=50, verbose=False)
self._model.eval()
x_test, y_test = test_set
--
GitLab