Final Version MS2 V2

2023-06-06 22:48:08 +02:00 · 2023-06-06 22:48:08 +02:00 · f0ebed4731
commit f0ebed4731
parent 50b71750ab
5 changed files with 2 additions and 2 deletions
--- a/ConvolutionalNeuralNetwork.py
+++ b/ConvolutionalNeuralNetwork.py
@ -66,7 +66,7 @@ model.compile(optimizer = 'Adam', loss = 'categorical_crossentropy', metrics = [
 history = model.fit(x = X_train_subset,
                    y = y_train_subset,
                    batch_size = 32,
-                    epochs = 5,
+                    epochs = 100,
                    verbose = 1,
                    validation_data = (X_valid_subset, y_valid_subset))

--- a/FullyConnectedNeuralNetwork.py
+++ b/FullyConnectedNeuralNetwork.py
@ -59,7 +59,7 @@ model.compile(optimizer = 'rmsprop', loss = 'categorical_crossentropy', metrics
 history = model.fit(x = X_train_subset,
                    y = y_train_subset,
                    batch_size = 32,
-                    epochs = 5,
+                    epochs = 500,
                    verbose = 1,
                    validation_data = (X_valid_subset, y_valid_subset))

--- a/saved_model/convolutionalNeuralNetwork.h5
+++ b/saved_model/convolutionalNeuralNetwork.h5
--- a/saved_model/fullyConnectedNeuralNetwork.h5
+++ b/saved_model/fullyConnectedNeuralNetwork.h5
--- a/saved_model/ownModel.h5
+++ b/saved_model/ownModel.h5