Skip to content
Snippets Groups Projects
Commit 75faede1 authored by johannes bilk's avatar johannes bilk
Browse files

fixed an error in the network test file

parent 80a40322
No related branches found
No related tags found
No related merge requests found
......@@ -8,6 +8,6 @@ numLayers = 3
numNeurons = 81, 81, 81
convolutions1 = 1
channels1 = 1,3
#channels12 = 1,5,5
#kernelSize1 = (3,3)
#kernelSize12 = (2,2)
channels12 = 1,5,5
kernelSize1 = (3,3)
kernelSize12 = (2,2)
......@@ -10,10 +10,9 @@ from machineLearning.nn.optim import SGD, SGDMomentum, NesterovMomentum, AdaGrad
from machineLearning.nn.scheduler import ExponentialLR, SteppedLR, CyclicalLR
from machineLearning.nn.module import Sequential, Parallel
from machineLearning.nn.loss import CrossEntropyLoss, MSELoss, NLLLoss, MAELoss, FocalLoss
from machineLearning.nn.observable import NetworkObservables
from machineLearning.data import Data
from machineLearning.utility import Time, Progressbar
from machineLearning.metric import ConfusionMatrix
from machineLearning.metric import ConfusionMatrix, Observables
from machineLearning.settings import NetworkSettings
......@@ -158,10 +157,16 @@ if __name__ == "__main__":
print(optim)
print(lossFunc)
# setting up training observables
metrics = Observables(settings['epochs'])
metrics.addOberservable('losses', 'descending')
metrics.addOberservable('validation', 'descending')
metrics.addOberservable('accuracy', 'ascending')
metrics.addOberservable('learningRate')
# Beginn training
print("Beginn training...")
timer.start()
metrics = NetworkObservables(settings['epochs'])
epochs = settings['epochs']
for i in range(settings['epochs']):
data.trainMode()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment