michellewidjaja commited on
Commit
0715c7f
1 Parent(s): e2a2d06

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Animal Faces CNN Model
2
+
3
+ This model is a CNN trained on the Kaggle Animal Face dataset to classify cats, dogs, and wildlife. It is based on the ResNet18 architecture.
4
+
5
+ ## Training Results:
6
+
7
+ - Validation Accuracy: 99.32%
8
+ - Dataset: Cats, Dogs, and Wildlife Faces
9
+ - Number of Epochs: 10
10
+
11
+ | **Epoch** | **Loss** | **Validation Accuracy** |
12
+ |-----------|----------|-------------------------|
13
+ | 1 | 0.2374 | 95.11% |
14
+ | 2 | 0.1716 | 95.86% |
15
+ | 3 | 0.1381 | 97.53% |
16
+ | 4 | 0.1250 | 97.48% |
17
+ | 5 | 0.1038 | 97.53% |
18
+ | 6 | 0.0867 | 97.61% |
19
+ | 7 | 0.0914 | 98.44% |
20
+ | 8 | 0.0656 | 99.09% |
21
+ | 9 | 0.0462 | 99.35% |
22
+ | 10 | 0.0432 | 99.32% |