diff --git a/NeuralNetwork/best_model.pth b/NeuralNetwork/best_model.pth index ef12ff6..209b131 100644 Binary files a/NeuralNetwork/best_model.pth and b/NeuralNetwork/best_model.pth differ diff --git a/NeuralNetwork/learning_results.txt b/NeuralNetwork/learning_results.txt index 70eda4b..7ad0f10 100644 --- a/NeuralNetwork/learning_results.txt +++ b/NeuralNetwork/learning_results.txt @@ -7,4 +7,25 @@ Epoch: 6 Train Loss: 0 Train Accuracy: 0.9545454545454546 Epoch: 7 Train Loss: 0 Train Accuracy: 0.972027972027972 Epoch: 8 Train Loss: 0 Train Accuracy: 0.9820179820179821 Epoch: 9 Train Loss: 0 Train Accuracy: 0.994005994005994 -Epoch: 10 Train Loss: 0 Train Accuracy: 0.9945054945054945 \ No newline at end of file +Epoch: 10 Train Loss: 0 Train Accuracy: 0.9945054945054945 + +Epoch: 1 Train Loss: 42 Train Accuracy: 0.6428571428571429 +Epoch: 2 Train Loss: 11 Train Accuracy: 0.8306693306693307 +Epoch: 3 Train Loss: 3 Train Accuracy: 0.8921078921078921 +Epoch: 4 Train Loss: 2 Train Accuracy: 0.8891108891108891 +Epoch: 5 Train Loss: 1 Train Accuracy: 0.9335664335664335 +Epoch: 6 Train Loss: 0 Train Accuracy: 0.952047952047952 +Epoch: 7 Train Loss: 0 Train Accuracy: 0.9545454545454546 +Epoch: 8 Train Loss: 0 Train Accuracy: 0.9655344655344655 +Epoch: 9 Train Loss: 0 Train Accuracy: 0.9815184815184815 +Epoch: 10 Train Loss: 0 Train Accuracy: 0.9805194805194806 +Epoch: 11 Train Loss: 0 Train Accuracy: 0.9855144855144855 +Epoch: 12 Train Loss: 0 Train Accuracy: 0.989010989010989 +Epoch: 13 Train Loss: 0 Train Accuracy: 0.9925074925074925 +Epoch: 14 Train Loss: 0 Train Accuracy: 0.9915084915084915 +Epoch: 15 Train Loss: 0 Train Accuracy: 0.9885114885114885 +Epoch: 16 Train Loss: 0 Train Accuracy: 0.994005994005994 +Epoch: 17 Train Loss: 0 Train Accuracy: 0.997002997002997 +Epoch: 18 Train Loss: 0 Train Accuracy: 0.9965034965034965 +Epoch: 19 Train Loss: 0 Train Accuracy: 0.999000999000999 +Epoch: 20 Train Loss: 0 Train Accuracy: 1.0 \ No newline at end of file diff --git a/NeuralNetwork/neural_network_learning.py b/NeuralNetwork/neural_network_learning.py index d3fb8e1..10b3a1e 100644 --- a/NeuralNetwork/neural_network_learning.py +++ b/NeuralNetwork/neural_network_learning.py @@ -21,7 +21,7 @@ optimizer = Adam(model.parameters(), lr=0.001, weight_decay=0.0001) #loss function criterion = nn.CrossEntropyLoss() -num_epochs = 10 +num_epochs = 20 # train_size = len(glob.glob(images_path+'*.jpg')) train_size = 2002 diff --git a/NeuralNetwork/older_best_model.pth b/NeuralNetwork/older_best_model.pth new file mode 100644 index 0000000..ef12ff6 Binary files /dev/null and b/NeuralNetwork/older_best_model.pth differ