Update README.md

README.md

@@ -35,3 +35,40 @@ by Daniel Glownia
 | Train  | 85,085  |
 | Test  | 2,500  |
 | Validation  | 2,500  |
+
+## CNN Implementation
+
+- MobileNetV3 as base model(transfer learning)
+- Trained on 100 epochs
+- Optimizer: Adam
+- Loss: Categorical Cross Entropy
+
+```python
+epochs = 100
+batch_size = 256
+
+inputs = pretrained_model.input
+x = processing_layers(inputs)
+
+x = Dense(256, activation='relu')(pretrained_model.output)
+x = Dropout(0.2)(x)
+x = Dense(128, activation='relu')(x)
+x = Dropout(0.2)(x)
+x = Dense(64, activation='relu')(x)
+x = Dropout(0.2)(x)
+
+
+outputs = Dense(500, activation='softmax')(x)
+
+model = Model(inputs=inputs, outputs=outputs)
+```
+
+## Results
+
+| Dataset  | Accuracy |
+| ------------- | ------------- |
+| Train  | 84.87%  |
+| Test  | 92.20%  |
+| Validation  | 93.36%  |
+
+![plot](loss.png)